1、下载各类包
清华源地址
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/?C=M&O=D
显卡驱动对应版本的cuda toolkit
个人显卡驱动为442.50,因此下载的是:
pytorch-1.5.1-py3.7_cuda102_cudnn7_0.tar.bz2
torchvision-0.7.0-py37_cu102.tar.bz2
下载成功之后放到anaconda3\pkgs文件夹下,也就是放各类包的地方
2、安装
conda install --offline pytorch-1.6.0-py3.7_cuda101_cudnn7_0.tar.bz2
conda install --offline torchvision-0.7.0-py37_cu101.tar.bz2
3、测试
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda_version)