在PyCharm中安装Pytorch的GPU版本

我在pycharm中已经加入了conda的环境了，所以直接就下载了pytorch的GPU版本，在pycharm里面训练跑项目是可以使用的。我试过了在conda的虚拟环境中和在哪个文件夹中使用没有关系。

一：下载pytorch

首先，进入pytorch的官网PyTorch，往下滑，找到这个地方

除了第三行，我在pycharm中使用的是pip这个命令。然后cuda则需要根据你电脑的版本来确定。

按住Win+R，输入cmd，打开命令行，输入

nvidia-smi

 会出现下面这个页面

CUDA Version：12.2代表你电脑最高能够安装的cuda版本，我们安装的cuda要低于这个版本。

然后输入

nvcc -V

查看我们自己电脑的cuda版本。

像我的就是11.8。查看完这些后，就可以根据下面的命令来安装Pytorch了。

二：验证是否安装成功

在Pycharm的Terminal中分别输入以下命令来检测是否安装成功。

python
import torch
torch.cuda.is_available()

如果输出为True，则说明安装成功了。