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

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

参考链接:
(1)https://blog.csdn.net/kellyroslyn/article/details/109668001
torch版本&cuda版本

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)

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

Guess you like

Origin blog.csdn.net/ahao6666666/article/details/113733621