pytorch环境安装

一、查看nvidia control Panel

由此知道了nvidia 驱动支持的cuda版本是12.2以下

二、安装cuda toolkit

在官网上下载对应版本的 toolkit
例如：

三、在anaconda中创建新环境、安装pytorch

创建新环境的步骤忽略；
创建完新环境后，安装pytorch
在链接处：https://pytorch.org/get-started/previous-versions/
找到合适的pytorch版本，找到对应的安装命令
例如：

推荐使用pip安装pytorch，速度很快
而且不会出现module not found等让人摸不着头脑的错误

四、验证pytorch

import torch
print(torch.__version__)
#验证pytorch是否正确安装

x = torch.rand(5, 3)
if torch.cuda.is_available():
    x = x.to("cuda")
    print(x)
else:
    print("CUDA is not available")
#验证 pytorch 是否可以使用GPU进行运算