--extra-index-url https://download.pytorch.org/whl/cu121 torch==2.1.2 torchvision numpy scipy scikit-image einops noise opencv-python pynvml wandb tensorboard cmake Pillow>=8.3.2 tqdm==4.35.0 wget cython lmdb av imutils imageio-ffmpeg qimage2ndarray albumentations requests==2.25.1 nvidia-ml-py3==7.352.0 pyglet rsa matplotlib