40系顯卡深度學習配置指南(pytorch)
眾所周知,每一代新顯卡都會與指定的cuda新版本唯一兼容,例如30系顯卡必須配合cuda11.X版本,40系也不例外,有獨特的cuda版本需求。
40系目前兼容的最早cuda版本為11.8,見以下截圖:

因此需要配置的cuda版本最早為11.8,下載的pytorch也需要與這個版本對應,官網的下載方式如下截圖:

截圖截錯了?我們看看另一邊的Preview呢:

好的,在preview里找到了。下載好如上版本的pytorch,cuda后即可配對,實現40系顯卡的深度學習配置環(huán)境。如果需要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
標簽: