Anaconda中PyTorch框架GPU/CPU版本安装

一、安装CUDA

CPU版本不需要安装CUDA。

CUDA官网地址：CUDA Toolkit 11.7 Update 1 Downloads | NVIDIA Developer

根据自己电脑的系统、版本等进行选择。我的是Windows 11 64位， 点击下载。

 将下载好的exe文件双击，按步骤操作，进行安装

二、安装PyTorch

完成CUDA的安装再来下载PyTorch.   PyTorch官方网站：PyTorch

同理，根据自己电脑情况，进行选取配置。若下载CPU版本，则自行选取CPU，CUDA表示GPU版本。其中可以通过Win+R,打开cmd,输入nvidia-smi，查看电脑CUDA版本。我的电脑是11.7版本，不过PyTorch没有该版本，我下的是11.6的，不过没有影响。

然后将上述红色框内的内容复制（把pop3改成pop）,粘贴到Anaconda Prompt中，回车进行下载。

 最后提示Successful,表示安装成功。打开IDE进行测试，看是否能导入包，输出PyTorch版本。

import torch
print(torch.__version__)

运行结果：

1.12.1+cu116

到此，PyTorch安装完成。