分享——安装使用gpu的pytorch的一些经历和总结
参考链接:
(1)https://blog.csdn.net/kellyroslyn/article/details/109668001
0 下载前,先将自己的镜像源改成清华大学镜像站
方法如下:https://blog.csdn.net/hanx09/article/details/105519266
1 安装torch的时候,先看自己电脑gpu的驱动版本,根据驱动版本找到cuda 的版本。
2 根据cuda的版本下载torch 的版本
3 下载以下三个库
4 我是在清华大学镜像站下载的.tar.bz2文件,然后手动安装(放在anaconda的pkgs文件里)
5 cmd进入这个目录(cd C:\ProgramData\Anaconda3\pkgs),然后conda install filename.tar.bz2即可(每个文件都需要conda install一下,包括cudatoolkit)
6 最后,如果import torch有问题的话,请更新一下numpy(conda update numpy)
如果还有其他问题欢迎指出。