个人 Windows 环境 pytorch+cuda 配置记录（2021.9）

最近重装了下工位电脑，需要重新配置编程环境

查看显卡对应的 CUDA版本

• 在桌面上右击 “NVIDIA 控制面板”
• 
• 在 “组件” 标签查看 driver 版本
• 
• 下载对应 CUDA 版本
• 下载链接：https://developer.nvidia.com/cuda-toolkit-archive
• 安装
• 在 cmd 中输入测试代码

nvcc -V

• 版本信息

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Mon_May__3_19:41:42_Pacific_Daylight_Time_2021
Cuda compilation tools, release 11.3, V11.3.109
Build cuda_11.3.r11.3/compiler.29920130_0

• 安装成功

安装 cudnn

• cudnn下载
• 官网下载：https://developer.nvidia.com/rdp/cudnn-archive
• 解压文件夹，将解压后的文件夹下的文件拷贝到cuda安装目录下，与之相对应的文件夹下
• 

安装对应 pytorch 版本

• 首先是 python 版本个人感觉 3.6 版本足矣，更高的版本不兼容低版本 pytorch
• 在官网下载对应版本 pytorch https://pytorch.org/get-started/locally/
• 或者低版本的：https://pytorch.org/get-started/previous-versions/
• 验证 cuda

>>> import torch
>>> torch.cuda.is_available()
True

• 成功