安装cuda

1.win+r打开终端，nvidia-smi查询显卡类型与版本号

2.根据cuda version，选择合适安装版本，例如，版本号是12.1，选择12.0及以下都可以

3.选择合适版本之后，cuda官网下载

https://developer.nvidia.com/cuda-toolkit-archive

4.安装过程中，第一个选择安装位置的页面不用管，直接点🆗，然后一直继续next

5.到了安装选项，选择自定义，组件自由选择，接着在此处选择安装位置的时候，就可以选择放在D盘了，但是一定要记住安装的位置

6.安装完成之后，设置系统变量，如图所示

7.win+r，cmd，输入nvcc -V，出现如下所示便是安装完成。

安装cudnn

1.进入cudnn网址，建议先注册登录再下载

https://developer.nvidia.com/rdp/cudnn-archive

2.选择和cuda版本合适的cudnn版本进行下载

3.下载好之后，是一个压缩包，解压，找到这三个文件

4.把这三个文件，复制粘贴到刚才安装cuda的文件夹里

5.验证cudnn是否安装完成，win+r，进行以下操作，（文件目录按照自己的来），分别运行，得到pass即代表安装成功

annaconda新建torch虚拟环境以及安装pytorch

1.打开conda prompt小黑窗，新建一个虚拟环境：conda create -n torch_env python=3.8

2.进入刚才新建的虚拟环境：conda activate torch_env

3.Start Locally | PyTorch找适配于你的cuda的pytorch版本

4.如果你不是最新的，点这里

5.进去往下翻，找到对应的cuda版本的pip安装命令，将该命令复制粘贴到刚才的conda prompt窗口，进行安装

6.安装完成后，使用pip list命令查看，是否有torch的几个包

7.在python编程软件中，选择刚才搞好的torch_env编译环境，输入以下代码

import torch

print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())

获得如下输出就是安装成功了

参考教程：

torch.cuda.is_available()返回false——解决办法_cuda available false-CSDN博客

最详细！Windows下的CUDA与cuDNN详细安装教程_cuda安装-CSDN博客

【超详细教程】2024最新Pytorch安装教程（同时讲解安装CPU和GPU版本）-CSDN博客