pytorch碎碎念

官网 (opens new window)

坑货安装步骤

参考doc (opens new window)

1. 检查电脑是否支持NVIDIA。

2. 如何安装CUDA。

   其中CUDA (opens new window)的版本不能超过本地GPU版本

3. 如何安装cudNN。

   cudNN (opens new window)的版本要和CUDA一致

4. 如何安装Pytorch。

5. 检查是否安装成功。

当前笔记本：

package	版本号
CUDA	12.4
cuDNN	9.1.0