pytorch与cuda的对应版本
时间: 2024-06-16 14:07:39 浏览: 124
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
PyTorch与CUDA的对应版本是为了在使用PyTorch时能够充分利用GPU加速计算的能力。具体的对应关系如下:
- PyTorch 1.0.x:支持CUDA 9.0
- PyTorch 1.1.x:支持CUDA 9.0、CUDA 10.0
- PyTorch 1.2.x:支持CUDA 9.0、CUDA 10.0、CUDA 10.1
- PyTorch 1.3.x:支持CUDA 9.2、CUDA 10.0、CUDA 10.1
- PyTorch 1.4.x:支持CUDA 9.2、CUDA 10.1、CUDA 10.2
- PyTorch 1.5.x:支持CUDA 9.2、CUDA 10.1、CUDA 10.2、CUDA 11.0
- PyTorch 1.6.x:支持CUDA 9.2、CUDA 10.1、CUDA 10.2、CUDA 11.0
- PyTorch 1.7.x:支持CUDA 9.2、CUDA 10.1、CUDA 10.2、CUDA 11.0
需要注意的是,PyTorch的版本和CUDA的版本需要匹配,否则可能会导致不兼容或者无法正常运行的问题。
阅读全文