tensorflow各版本支持的CUDA和CUDNN版本

最新推荐文章于 2024-05-31 09:10:14 发布

Samuel'C

最新推荐文章于 2024-05-31 09:10:14 发布

阅读量8.1k

点赞数 6

文章标签： linux

本文链接：https://blog.csdn.net/m0_47029116/article/details/118698209

版权

博客
CUDA：用于GPU编程的语言，跑TF的时候用了GPU，TF里面很多函数（或者依赖库）是CUDA语言编写的。不同TF版本需要不同的CUDA。

CUDNN：NVIDIA为深度学习，矩阵运算写的一个加速库。CUDA版本必须和cuDNN版本匹配。cuDNN和TF版本匹配与否则无所谓（不过CUDA和TF版本必须匹配，所以cuDNN版本和TF版本是间接依赖关系）

TF：这个没什么好说的。个人经验，别用太新的，如果每次都用最新的，那么进而导致CUDA也要不断更新

CUDA驱动版本：保持最新就好了，CUDA驱动版本要求和CUDA版本匹配，而CUDA又要求cuDNN/TF是匹配的。不过CUDA驱动版本是向下兼容的，所以保持最新就没事。输入nvidia-smi可以查看：

关于怎么样查看本地CUDA \ CUDNN \ TF 版本的常用命令可以见我的这篇文章常用的Linux指令（包含基础指令，CUDA和Python相关）

版本	Python 版本	编译器	构建工具
tensorflow-2.4.0	3.6-3.8	GCC 7.3.1	Bazel 3.1.0
tensorflow-2.3.0	3.5-3.8	GCC 7.3.1	Bazel 3.1.0
tensorflow-2.2.0

关注