Pytorch1.5.1 cuda10.2版本python3.7安装方法

安装步骤

1.下载pytorch。在清华源https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/?C=M&O=D下载对应pytorch版本，pytorch-1.5.1-py3.7_cuda102_cudnn7_0.tar.bz2。

2.下载torchvision，torchvision===0.6.1。

3.进入下载路径进行离线安装。打开conda prompt,输入

cd E:\data\torch_bag #改成自己的下载路径

离线安装，

conda install --offline pytorch-1.5.1-py3.7_cuda102_cudnn7_0.tar.bz2

conda install --offline torchvision-0.6.1-py37_cu102.tar.bz2

4.安装cudatoolkit=10.2

conda install cudatoolkit=10.2

5.检验是否安装成功

import torch
from torch.backends import cudnn

x = torch.Tensor([1.0])
xx = x.cuda()
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
print(xx)
print(cudnn.is_acceptable(xx))

参考链接：http://www.qb5200.com/article/226179.html