torch2.0.1支持cuda12.1吗
时间: 2024-05-28 13:14:44 浏览: 234
CUDA10.1向下兼容10.0,完美运行官方TensorFlow、Pytorch_pytorchcuda10.0,cuda10.0可以升级
5星 · 资源好评率100%
不好意思,目前(截至2021年12月)PyTorch的最新版本是1.10.0,官方支持的CUDA版本范围为CUDA 10.2、CUDA 11.1、CUDA 11.3和CUDA 11.4,暂时还不支持CUDA 12.1。如果你使用的是CUDA 12.1,可能需要等待PyTorch官方发布支持该版本的新版本,或者尝试在非官方渠道寻找相关支持。
阅读全文