pytorch与cuda版本对应关系汇总

  1. pytorch与cuda版本关系
cuda版本 支持pytorch版本
cuda10.2 1.5 ~ 1.12
cuda11.0 1.7 ~ 1.7.1
cuda11.1 1.8 ~ 1.10.1
cuda11.3 1.8.1 ~ 1.12.1
cuda11.6 1.12.0 ~ 1.13.1
cuda11.7 1.13.0 ~ 2.0.1
cuda11.8 2.0.0 ~ 2.1.1
cuda12.1 2.1.0 ~ 2.1.1
  1. cuda 与 cudnn关系
cuda版本 支持cudnn版本
cuda10.2 v7.6.5 ~ v8.7.0
cuda11.3 v8.2.0 ~ v8.9.6
  1. pytorch 与 python关系
torch torchvision Python
main / nightly main / nightly >=3.8, <=3.11
2.0 0.15 >=3.8, <=3.11
1.13 0.14 >=3.7.2, <=3.10
1.12 0.13 >=3.7, <=3.10
1.11 0.12 >=3.7, <=3.10
1.10 0.11 >=3.6, <=3.9
1.9 0.10 >=3.6, <=3.9
1.8 0.9 >=3.6, <=3.9
1.7 0.8 >=3.6, <=3.9
1.6 0.7 >=3.6, <=3.8
1.5 0.6 >=3.5, <=3.8
1.4 0.5 ==2.7, >=3.5, <=3.8
1.3 0.4.2 / 0.4.3 ==2.7, >=3.5, <=3.7
1.2 0.4.1 ==2.7, >=3.5, <=3.7
1.1 0.3 ==2.7, >=3.5, <=3.7
<=1.0 0.2 ==2.7, >=3.5, <=3.7

相关推荐

  1. pytorchcuda版本对应关系汇总

    2023-12-28 12:50:06       31 阅读
  2. Android 平台代码、版本API级别对应关系

    2023-12-28 12:50:06       50 阅读
  3. Android中 Gradle AGP 版本对应关系

    2023-12-28 12:50:06       29 阅读

最近更新

  1. TCP协议是安全的吗?

    2023-12-28 12:50:06       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2023-12-28 12:50:06       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2023-12-28 12:50:06       19 阅读
  4. 通过文章id递归查询所有评论(xml)

    2023-12-28 12:50:06       20 阅读

热门阅读

  1. MySQL数据库编译及安装

    2023-12-28 12:50:06       36 阅读
  2. salesforce role和profile的区别

    2023-12-28 12:50:06       32 阅读
  3. 【网络协议】WebSocket知识点梳理和总结

    2023-12-28 12:50:06       27 阅读