Python 环境配置（三）安装pytorch

Python 环境配置(三)安装pytorch

一、CUDA

安装CUDA视安装的Pytorch版本而定，GPU版本 需要 安装CUDA，CPU版本 无需 安装CUDA。

1、查看驱动版本

方法一：

---

方法二：

---

2、下载

CUDA Toolkit Archive | NVIDIA Developer

选择对应的版本，对应的！！！

此后依次选择下载即可

注意 windows的版本不要选错

---

3、安装

临时抽取文件放置位置的， 此处直接点击ok

没有安装vs，直接安装的这个，需要把CUDA里面的 Visual Studio Integration 取消勾选，否则会安装不成功

更改位置

---

4、验证

nvcc -V

---

二、cuDNN

cuDNN 9.3.0 Downloads | NVIDIA Developer

把这三个文件夹的文件分别拷贝到CUDA安装目录对应的(bin、include、lib)文件夹中

三、Pytorch

以下仅介绍GPU版本，CPU版本请参考：Pytorch最全安装教程(一步到位)_pytorch安装-CSDN博客

1、官网找语句

PyTorch
点击 Get start

---

2、下载

---

可以在下载之前 把 魔法 打开。。。会更快些(效果很明显)

---

Anaconda Powershell Prompt 里 输入对应的命令

时间有点长，耐心等待。。。

如果异常停止，则重新输入命令 来继续下载

---

3、测试

Anaconda Powershell Prompt 里 输入对应的命令

python


import torch
 
print(torch.__version__)
print("gpu", torch.cuda.is_available())

如图，即成功。