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))