前言

本人显卡设备是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

1
2
3
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