torch==1.7.1 torchvision==0.8.2 CUDA==10.2 cudnn==10.2
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
conda install cudatoolki=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
conda install cudnn=8.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html