Anaconda环境下安装gpu版pytorch

cuda安装

首先到下面的网址下载cude，注意，不要下载最新的，目前pytorch支持的最新版本是11.8。
https://developer.nvidia.com/cuda-toolkit-archive

pytorch安装

打开Anaconda自带的命令行，如下图所示。再到下面的网站获取安装命令。
https://pytorch.org/get-started/locally/


在安装前需要在命令行中创建一个新的conda环境，以使用GPU版本的PyTorch。下面代码创建了一个新的名为myenv的conda环境，并激活到了myenv。

conda create --name myenv
conda activate myenv

激活新的conda环境myenv后，输入上图中得到的安装pytorch的命令。

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

测试

在激活新的conda环境myenv后，输入以下指令，输出为True即成功。

python -c "import torch; print(torch.cuda.is_available())"

在pycharm中选择conda环境