Windows系统下的PyTorch安装

安装pytorch

这一块的教程很多，我这里给出自己的例子。
下表是本人电脑有关cuda的配置信息

显卡	RTX 3060
NVIDIA-SMI	527.99
Driver Version	527.99
CUDA Version	12.0
Cuda compiler driver	11.3

这里CUDA Version12.0的意思是显卡支持的最高CUDA的版本，本人选择使用的是11.3版本的，原因在于该版本相对比较稳定。
确认电脑完成CUDA的配置后，开始进行pytorch的安装

1. 进入pytorch官网：PyTorch官网
   根据电脑配置选择对应的版本，本人选择如下：
   
   获得的命令如下：

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

2. 在激活后的环境中运行安装
   
3. 测试是否成功安装
   控制台命令窗口输入python，导入torch并查看torch的版本，如下图所示：
   
   安装成功！

注意

其他版本的pytorch如果无法正确选择合适的版本获得命令，可以尝试在PyTorch官网选择使用pip3安装，然后根据可支持的cuda版本修改下图红线中的链接。红线末尾的cu117意思是CUDA 11.7版本，因此可根据个人电脑可支持的cuda版本进行更改，比如电脑最高支持CUDA 11.3，就可以把cu117修改为cu113，那么命令就变为：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113

你甚至可以访问该链接：https://download.pytorch.org/whl/cu113下载对应的包进行本地安装。