支持pytorch11.8的cuda
时间: 2023-09-29 08:04:12 浏览: 123
【环境配置】RTX 3090+cuda11.2+python39+pytorch1.8.0(csdn)————程序.pdf
5星 · 资源好评率100%
目前最新版本的 PyTorch(1.9.0)支持 CUDA 11.1 和 11.2。如果您需要支持 CUDA 11.0 或 11.3,可以使用 PyTorch 1.8.0 或 1.10.0 的相应版本。如果您要使用 CUDA 11.4,需要等待 PyTorch 1.10.0 发布。所以,要支持 CUDA 11.8,您需要等待 PyTorch 更新并支持该版本的 CUDA。
阅读全文