CUDA Toolkit 11.7、11.8与12.1官方下载地址
💡
原文中文,约1300字,阅读约需3分钟。
📝
内容提要
本文提供了CUDA Toolkit 11.7、11.8和12.1的官方下载地址,以及与PyTorch的兼容性。同时,还包括了cuDNN的下载地址和解决CUDA错误的方法。
🎯
关键要点
- CUDA Toolkit 11.7、11.8与12.1的官方下载地址分别为:11.7 - https://developer.nvidia.com/cuda-11-7-0-download-archive,11.8 - https://developer.nvidia.com/cuda-11-8-0-download-archive,12.1 - https://developer.nvidia.com/cuda-downloads。
- PyTorch与CUDA版本的兼容性:PyTorch2.0.1支持CUDA 11.7和11.8,PyTorch Preview支持CUDA 11.8和12.1。
- 重装PyTorch时可以使用命令:pip3 install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --no-cache。
- cuDNN的官方下载地址为:https://developer.nvidia.com/rdp/cudnn-download,需登录NVIDIA开发者账号。
- 如果出现CUDA错误:CUDA Error: no kernel image is available for execution on the device,可能是显卡型号较老,需要手动编译PyTorch。
➡️