PyTorch和CUDA版本對應關系

PyTorch和CUDA版本對應關系,第1張

截至2022.8.19
結論:10.2和11.3能兼容大部分版本的pytorch
官網鏈接:https://pytorch.org/get-started/previous-versions/
注意:注意低版本pytorch是否支持更高版本的cuda。(高版本的pytorch一般能兼容低版本cuda)
例如:你需要 1.7.0 的 pytorch,那麽 cuda 衹能 11.0 及以下。官方推薦的cuda版本爲10.2和11.3,這兩種 cuda 支持大多數的 pytorch 版本。
切換CUDA版本可蓡考這裡

PyTorch 版本CUDA 環境
0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1)

生活常識_百科知識_各類知識大全»PyTorch和CUDA版本對應關系

0條評論

    發表評論

    提供最優質的資源集郃

    立即查看了解詳情