PyTorch和CUDA版本對應關系


截至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) 9.2
1.2.0、1.1.0、1.0.0(1) 10.0
1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 10.1
1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1) 10.2
1.7.0(1) 11.0
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0 11.1
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1) 11.3
1.11.0 11.5
1.12.0(1) 11.6


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM