Pytorch-gpu的安装_pytorch gpu安装

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())
1
2
3

4.在一个新的python环境中安装pytorch（选读）

怎样在pycharm中配置自己创建的环境，可以参考我这篇文章：
pycharm配置创建的python环境
怎样用anaconda创建一个环境，以及与环境相关的种种命令，可以参考我这篇文章：
anaconda虚拟环境常用命令