前言

本人显卡设备是RTX2080 SUPER, 其他版本显卡不一定适用，仅供参考。

版本

1. cuda: cuda_11.8.0_522.06_windows.exe
2. cudnn: cudnn-windows-x86_64-8.6.0.163_cuda11-archive
3. Anaconda3: Anaconda3-2023.09-0-Windows-x86_64

环境变量

anaconda3

1. C:\ProgramData\anaconda3
2. C:\ProgramData\anaconda3\Scripts
3. C:\ProgramData\anaconda3\Library\bin
4. C:\ProgramData\anaconda3\Library\mingw-w64\bin

CUDA环境变量

1. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib
2. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include
3. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\CUPTI\lib64

安装步骤

1. 安装anaconda3
2. 设置anaconda3环境变量，以及源.condarc文件
3. 创建虚拟环境，conda create --name [虚拟环境名] python=3.8
4. 安装基础依赖：pip install -r requirements.txt
5. 查看torch是否为GPU版本，不是则删除重新安装pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
6. 安装cuda
7. 设置cuda环境变量
8. 将cudnn的lib、include和bin文件夹拷贝到cuda的安装目录中

Torch查看是否支持GPU

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

安装包资源

1. cuda: https://mypikpak.com/s/VNuSLnwNwX03zmiK2OkdDjz4o1
2. cudnn: https://mypikpak.com/s/VNuSM8-CSkNS3vSpiam7JEvEo1
3. Anaconda3: https://mypikpak.com/s/VNuSM2U1SkNS3vSpiam7JCUco1