40系显卡深度学习配置指南(pytorch)
众所周知,每一代新显卡都会与指定的cuda新版本唯一兼容,例如30系显卡必须配合cuda11.X版本,40系也不例外,有独特的cuda版本需求。
40系目前兼容的最早cuda版本为11.8,见以下截图:

因此需要配置的cuda版本最早为11.8,下载的pytorch也需要与这个版本对应,官网的下载方式如下截图:

截图截错了?我们看看另一边的Preview呢:

好的,在preview里找到了。下载好如上版本的pytorch,cuda后即可配对,实现40系显卡的深度学习配置环境。如果需要cudnn的同学,下载cudnn对应的11.x版本即可。
cuda下载直链:https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_522.06_windows.exe
pytorch下载直链(pip格式):pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118