GPU版本Pytorch安装教程

1.首先Nvidia官网下载cuda

https://developer.nvidia.cn/cuda-downloadshttps://developer.nvidia.cn/cuda-downloads

根据自己电脑选择 我这里是Windows11 点击下载

下载完成之后完成安装即可。

然后打开Aonconda 里的Prompt命令窗口

 使用conda创建虚拟环境

 使用如下命令 其中xxx是你的环境名自定义 后面是python版本

conda create -n xxx python=x.x

 然后激活环境 xxx为环境名

conda activate xxx

 然后去pytorch官网下载pytorch

 PyTorch

复制命令完成安装：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

这样GPU版本的pytorch就已经创建完成了

可以使用我们创建的环境新建一个project 输入以下命令判断GPU是否可用

 
import torch 
print(torch.cuda.is_available())

返回True证明已成功安装。