windows cuda10.2 cudnn7.6.5下安装pytorch1.7.1 torchvision-0.8.2
pytorch版本的安装和cuda的版本是又对应关系的,如果版本不对应,是无法使用GPU进行工作的win+r.输入cmd打开终端输入:nvcc-V或者(setcuda)可以查看自己的cuda版本,我这里安装的是cuda10.2,cudnn安装的为7.6.5此环境下直接安装pytorch1.7.1,要么安装不成功,报没有pytorch1.7.1的版本,要么安装的是cpu版本的pytorch,可以通过