Install.md

Installation

(pytorch 1.11/13 or other versions also work)

The project is built with PyTorch 3.8, PyTorch 1.8.1. CUDA 10.2, cuDNN 7.6.5 For installing, follow these instructions:

conda install pytorch=1.8.1 torchvision=0.9.1 -c pytorch
pip install tensorboard einops scikit-image pytorch_msssim opencv-python

Please use the pillow package downloaded by Conda rather than pip.

Install warmup scheduler:

cd pytorch-gradual-warmup-lr/
python setup.py install
cd ..