CUDA11.2支持的pytorch版本
时间: 2024-01-08 11:04:14 浏览: 163
【环境配置】RTX 3090+cuda11.2+python39+pytorch1.8.0(csdn)————程序.pdf
5星 · 资源好评率100%
CUDA 11.2 支持的 PyTorch 版本是 1.8.1。你可以在 PyTorch 官方网站上查看支持不同 CUDA 版本的 PyTorch 版本列表。注意,支持 CUDA 11.2 的 PyTorch 版本需要使用对应的 CUDA 驱动程序和 cuDNN 版本。你可以在 NVIDIA 官方网站上下载相应的驱动程序和 cuDNN 版本。
阅读全文