windows cuda10.2 cudnn7.6.5下安装pytorch1.7.1 torchvision-0.8.2

pytorch版本的安装和cuda的版本是又对应关系的,如果版本不对应,是无法使用GPU进行工作的

win+r.输入cmd打开终端输入:nvcc -V或者(set cuda)可以查看自己的cuda版本,我这里安装的是cuda10.2,cudnn安装的为7.6.5

此环境下直接安装pytorch1.7.1,要么安装不成功,报没有pytorch1.7.1的版本,要么安装的是cpu版本的pytorch,可以通过
python
import torch
print(torch.version)的方式输出torch的版本是否是cpu版本的,如果是的话会输出torch 1.7.1+cpu,如果只输出的torch则人为安装的为GPU版本的,然后通过print(torch.cuda.is_available()),如果输出为True,表示安装成功,可以使用GPU工作,如果返回为false,则依然存在问题

pytorch安装
为避免上述问题,可以通过下载whl安装包的方式安装,链接https://download.pytorch.org/whl/torch_stable.html
windows cuda10.2 cudnn7.6.5下安装pytorch1.7.1 torchvision-0.8.2_第1张图片

根据cuda,python,系统版本去选择对应的pytorch版本和torchvision0.8.2版本,通过一下命令安装
pip install torch-1.7.1-cp36-cp36m-win_amd64.whl
pip install torchvision-0.8.2-cp36-cp36-win_amd64.whl

完成后,再次直行下面的命令进行查看
python
import torch
print(torch.version)
print(torch.cuda.is_available())
安装完成

你可能感兴趣的:(windows,pytorch,深度学习)