一、cuda安装
下载地址：cuda下载官网链接
1.1、cuda版本选择
这里有个前置工作需要搞清楚的，就是自己的显卡支持的cuda版本。
点击电脑左下角开始菜单找到
点击【帮助】下【系统信息】
点击【组件】：
即可查看显卡适配的cuda版本,要<=CUDA Driver Version。此处即<=12.0。
博主安装的是11.7。
1.2、下载安装
这里选择默认目录即可，是个安装过程的临时目录，结束后此文件夹会被自动删除。需要注意的是这里并非指定安装目录，若需要指定安装目录，此处需与指定安装目录不同。
扫描二维码关注公众号，回复：
14648983 查看本文章
二、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是否成功安装。
Python
import torch
torch.cuda.is_available()
