--extra-index-url https://download.pytorch.org/whl/cu118

jupyter
notebook
matplotlib
numpy
torch
docker
scikit-learn
seaborn
tqdm
einops
torchdiffeq 
timm
openmim
gdown
faiss-cpu
nvidia-ml-py3
pynvml