pytorch与cuda版本对应关系汇总

2023-12-27 18:44:58
  1. pytorch与cuda版本关系
cuda版本支持pytorch版本
cuda10.21.5 ~ 1.12
cuda11.01.7 ~ 1.7.1
cuda11.11.8 ~ 1.10.1
cuda11.31.8.1 ~ 1.12.1
cuda11.61.12.0 ~ 1.13.1
cuda11.71.13.0 ~ 2.0.1
cuda11.82.0.0 ~ 2.1.1
cuda12.12.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 |

文章来源:https://blog.csdn.net/u011489887/article/details/135250561
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。