实验环境为RTX 3060的Windows 11。我在一开始没有安装合适的CUDA版本,导致最终Pytorch训练时报错,最终只能推倒重来,浪费了不少时间。
为了提醒更多的同学在一开始就安装正确的版本,在此记录。
经实验,完全没问题的版本是:
1. CUDA Toolkit 11.1.1
CUDA Toolkit 11.1 Update 1 Downloads | NVIDIA Developer
依次选择:
2. cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2
cuDNN Archive | NVIDIA Developer
依次选择: