分享——安装使用gpu的pytorch的一些经历和总结

分享——安装使用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)

如果还有其他问题欢迎指出。