当前位置:   article > 正文

pytorch版本和cuda版本对应_pytorch和cuda对应版本

pytorch和cuda对应版本

pytorch版本和cuda版本对应

pytorch官网

高版本的pytorch一般能兼容低版本cuda。
例如:你需要 1.7.0 的 pytorch,那么 cuda 只能 11.0 及以下。官方推荐的cuda版本为10.2和11.3,这两种 cuda 支持大多数的 pytorch 版本。

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.010.2
1.7.0(1)11.0
1.8.0(1)、1.9.0、1.10.011.1
1.8.0(1)、1.9.0、1.10.011.3
声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号