1.先安装cuda和cudnn
推荐安装cuda11.3和cuda10.2,因为这两个版本用的多。
安装教程可以看我的另一篇文章:
cuda和cudnn安装教程
2.安装对应版本的Pytorch
Pytorch的版本需要和cuda的版本相对应。具体对应关系可以去官网查看。这里先附上一张对应关系图。
比如我的cuda是11.3的,可以下载的pytorch版本就有1.12.1,1.12.0,1.11.0等等。
确定好要下载的版本后,进入pytorch官网开始下载。Pytorch官网
我选择的是pytorch1.12,复制红线划中的内容到anaconda prompt中,就开始下载啦!
3.测试安装是否成功
在pycharm中输入下面代码,输出结果为true则说明安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
4.在一个新的python环境中安装pytorch(选读)
怎样在pycharm中配置自己创建的环境,可以参考我这篇文章:
pycharm配置创建的python环境
怎样用anaconda创建一个环境,以及与环境相关的种种命令,可以参考我这篇文章:
anaconda虚拟环境常用命令