安装环境pytorch
Previous PyTorch Versions | PyTorch
虚拟环境中安装cuda和cudnn
conda search cudatoolkit --info
找见cuda10.2
下载、安装。
conda install --use-local "D:\download\cudatoolkit-10.2.89-h74a9793_0.conda"
conda search cudnn --info
conda install --use-local "D:\download\cudnn-7.6.5-cuda10.2_0.conda"
pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu102