Pytorch、Tensorflow与CUDA的版本对应

最新推荐文章于 2024-08-16 09:41:32 发布

nnzzll

最新推荐文章于 2024-08-16 09:41:32 发布

阅读量1.5k

点赞数 1

分类专栏： pytorch 文章标签： tensorflow 深度学习人工智能 pytorch cuda

本文链接：https://blog.csdn.net/weixin_42694889/article/details/115509413

版权

pytorch 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

本文详细列举了CUDA不同版本与PyTorch和Tensorflow的兼容性情况。PyTorch相对宽容，支持较低版本的CUDA，而Tensorflow对CUDA版本要求严格，甚至不支持CUDA 10.2。对于遇到问题的用户，文中提到了Linux系统下通过symlink解决CUDA 10.2兼容性的方法，并提供了官方安装源链接。

摘要由CSDN通过智能技术生成

cuda	Pytorch	Tensorflow
10.0	1.0.0~1.4.0	1.13.0~2.0.0
10.1	1.3.0~1.8.1	2.1.0~2.3.0
10.2	1.5.0~1.8.1	无
11.0	1.7.0~1.7.1	2.4.0
11.1	1.8.0~1.9.0	无
11.2	无	2.5.0

更新于2021-09-18
仅记录cuda10.0以上版本
Pytorch似乎可以支持低版本的cuda，Tensorflow对版本要求更严格，必须要有对应版本的cudart64_xx.dll（xx为版本号）
吐槽一下辣鸡的Tensorflow，竟然直接就不支持cuda10.2了，在官方github的issue中有不少人提到这个问题，官方的回答就是让你去下旧版本或者新版本（呕
接上，也有大佬找到了Linux系统下将cuda10.2 symlink 到10.1的方法，见issue链接、解决教程

参考链接