anaconda环境安装pytorch(实际上还是pip安装的)

1.前置条件

安装了cuda和cudnn，其中cudnn是压缩版

1、安装cuda和cuDNN

首先系统支持的cuda，打开nvidia控制面板，点击左下方的“系统信息”，点击“组件”，显示版本是“12.1.112”。

到cuda官网下载相应版本，一般是向下兼容（低一点的版本是可以的）

kCUDA Toolkit Archive | NVIDIA Developer

网上很多下载及安装教程，这里不累赘。但是有个tips:第一次安装，以下页面建议选择“自定义”选项（借用大神的图）

自定义选项里不要勾选“Visual Studio”（前面步骤没有装的话，这又是一个tips：有人建议先到微软官网安装Visual Studio，但是后来发现不安装也能用）

安装完成后，大部分需要手动添加环境变量，方便检测是否安装成功。

win11是在桌面右键，点击“显示设置”，点击左边选择“系统”选项，选择右边的“系统信息”选项，“高级系统设置”-“高级”-“环境变量”-“系统变量”-”path”-“编辑”-“新建”，把相关内容加进去。（一共四个环境变量，一般是自动生成2个cuda_path和cuda_path_v12.1，另外两个手动添加）

至此，cuda安装完成，启动cmd，输入nvcc –version可以查看版本号，输入set cuda可以查看cuda设置的环境变量。

安装cuDNN。cuDNN不是一个安装软件，就是三个最新的文件夹，替代cuda里面的文件，下载，复制，粘贴到cuda安装目录即可。不装cuDNN也可，只是性能没那么好。