pytorch gpu版本安装

Anaconda Prompt中创建一个名为pytorch的环境，环境中用到的python版本为3.8

conda create -n pytorch python=3.8

如图所示

输入conda activate pytorch激活到pytorch环境

conda activate pytorch

查看cuda版本如果没有安装可以 查看下载cuda和cudnn（11.8版本）

pytorch

由于我安装的是cuda11.8版本，所以选择的是对应版本

如果安装的cuda版本低的话可以点击

可以根据自己下载的cuda版本进行选择如果没有显卡或者没有下载cuda可以选择 cpu版本的

复制命令到刚才激活的pytorch环境中

等待安装成功

验证pytorch是否成功

验证pytorch是否安装成功

python
import torch
torch.__version__
torch.cuda.is_available()

输入上述代码，成功如图所示