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
复制命令完成安装:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
这样GPU版本的pytorch就已经创建完成了
可以使用我们创建的环境新建一个project 输入以下命令判断GPU是否可用
import torch
print(torch.cuda.is_available())
返回True证明已成功安装。