安装目录

• 一、cuda安装
• • 1.1、cuda版本选择
  • 1.2、下载安装
• 二、cudnn安装
• 三、pytorch安装

一、cuda安装

下载地址：cuda下载官网链接

1.1、cuda版本选择

这里有个前置工作需要搞清楚的，就是自己的显卡支持的cuda版本。

点击电脑左下角开始菜单找到

点击【帮助】下【系统信息】

点击【组件】：

即可查看显卡适配的cuda版本,要<=CUDA Driver Version。此处即<=12.0。

博主安装的是11.7。

1.2、下载安装

这里选择默认目录即可，是个安装过程的临时目录，结束后此文件夹会被自动删除。需要注意的是这里并非指定安装目录，若需要指定安装目录，此处需与指定安装目录不同。

二、cudnn安装

下载地址：cudnn官网下载

选择适配所选cuda版本以及系统的cudnn下载。

下载后解压。直接将其4个文件拷入到上述cuda的安装目录下：

注意直接复制粘贴即可，会自动合并。

查看是否安装成功：打开cmd终端，输入命令行：

nvcc -V

输出如下信息即表示cuda安装成功。

测试cudnn是否安装成功：测试目录D:\CUDA11.7\extras\demo_suite目录下两个exe文件后是否出现跑“pass”，出现即为安装成功。

三、pytorch安装

下载地址：pytorch官网下载

查看适配cuda的pytorch版本。

如有需要可以新建虚拟环境后安装pytorch：

conda create -n your_env_name python=3.8

创建好之后激活虚拟环境：

conda activate your_env_name

可以看到左边的环境已经切换为新建的“jpytorch”。

博主的是cuda11.7，使用如下命令即可安装:

如果中途因为联网失败错误可以直接再次执行安装代码，安装完成后会出现“done”字样。

最后进入Python环境中测试torch是否成功安装。

Pythonimport torchtorch.cuda.is_available()