pytorch==1.9.1
cudatoolkit==11.1
torchvision
numpy
scipy
scikit-image
Pillow==6.1
setuptools==59.5.0
einops
noise
opencv-python
pynvml
wandb
tensorboard