conda install pytorch=1.8.1 torchvision=0.9.1 -c pytorch
pip install tensorboard einops scikit-image pytorch_msssim opencv-python
Install warmup scheduler:
cd pytorch-gradual-warmup-lr/
python setup.py install
cd ..