windows8.1+pytorch+cuda+cudnn
过了一阵子 感觉没有CUDA干不了活
重新下了CUDA和cuDNN。cuda和cudnn下载较慢 可复制链接到迅雷下载 cuDNN我win8.1 下载的win7
CUDA安装的坑
试了n多遍安装
按照很多博客写的 安装中自定义 安装,只勾选一部分。安装都ok 测试nvcc -V 也ok 但就是出现 控制面板中NAVIA打不开了!!巨坑!
最后解决办法为,将nvcc卸载干净,重新装下载的,这次选择精简,全都装了,装好后控制面板能打开,nvcc -V也木问题了,感动
安装pytorch
此处也踩坑巨多,在官网中复制命令pip 和conda都试过,太慢,用从打下载加上镜像源速度ok,但装的pytorch,似乎不是CUDA的,仍为cpu的,最终妥协,用下载好的whl文件安装好了
下载pytorch 后安装 我用的CUDA9.0 python3.7 对应找版本
下载地址https://pytorch.org/get-started/previous-versions/
pip安装参看https://blog.csdn.net/caracol/article/details/94832202
cmd中输入 pip +下载地址即可,参照图
下载whl后pip安装 ok之后
print(torch.cuda.is_available())显示true 完成!!!
安装torchvision
安装torchvision 应该要对应版本 但没有找到CUDA9.0 torch0.4.1对应版本,根据上一次安装经验conda install torchvison和 conda search torchvison 都找不到torchvision
同样去官网下载
下载前者(刚开始下载的后者,安装中报错找不到DLL)然后pip install +whl
随后测试,import torchvision 提示缺少PIL
缺少pillow,简单 conda install pillow 再import ok!!
最后检验,终于成功,感恩~