实际工作当中，经常维护好几个项目的代码，不同项目依赖的TF版本不一致问题。网上找了好多资料，但是每次遇到的问题都不一样，每次都要去查（就是是一样的问题，解决办法也可能会不一样）每次踩坑无数，今天痛定思痛，下决心总结一下。

基本概念

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驱动版本

工作当中改如何做？

尽量保持最新的CUDA驱动版本
维护多个cuda版本：cuda安装到/usr/local/目录下，可以通过命令切换不同版本

sudo rm -rf /usr/local/cuda #删除之前创建的软链接
sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda #创建新 cuda 的软链接

将cuDNN和CUDA解耦：说白了就是，二者单独存放，不要混在一起。切换cuDNN只需要在~/.bashrc文件里面设置如下即可：

# 在.bashrc文件里面写入该环境变量，切换cudnn只需要把cudnn设置到/root/cuda路径即可
export LD_LIBRARY_PATH=/root/cuda/lib64:$LD_LIBRARY_PATH

TF版本别用太新的：使用pip install tensorflow-gpu=1.x.0安装
遇到问题不要无脑google：先自行分析原因，尝试办法，然后再google。

关于CUDA，cuDNN，TF，CUDA驱动版本兼容问题相关推荐

矩池云conda创建指定cuda cudnn tf版本的虚拟环境
本文将介绍如何在矩池云通过conda创建虚拟环境,并安装指定版本的cuda cudnn tf. 需求:在矩池云中安装一个虚拟环境,python3.6,tensorflow-gpu1.4,cuda8,c ...
不同版本cuda对应的NVIDIA驱动版本
不同版本cuda对应的NVIDIA驱动版本引用:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 运行CUDA应用 ...
Linux服务器安装cuda,cudnn，显卡驱动和pytorch超详细流程
原文链接:https://blog.csdn.net/kingfoulin/article/details/98872965 基本的环境首先了解自己服务器的操作系统内核版本等信息: 查看自己操作系统 ...
CUDA对应的NVIDIA驱动版本对照表
一般而言,不同版本的CUDA要求不同的NVIDIA驱动版本,同时显卡驱动版本要不低于CUDA的安装版本,具体的对照关系如下: 如下链接对应了官方的版本要求说明: https://docs.nvidia ...
cuda安装、gcc多版本兼容安装切换
安装cuda开发库之后,使用gcc编译会可能会报错要求gcc版本,如下表 CUDA version max supported GCC version 11.1, 11.2, 11.3 10 11 9 ...
nvidia显卡驱动、cuda、cudnn、tensorflow对应版本
1.下载显卡驱动在nvidia官网下载驱动,驱动官网选择设备的驱动进行搜索下载即可,搜索时注意对应的操作系统一般为安装NVIDIA Studio驱动版本,GeForce Game Ready适用于 ...
一篇文章清晰了解NVIDAI显卡驱动（包括：CUDA、CUDA Driver、CUDA Toolkit、CUDNN、NCVV）
背景开发过程中需要用到GPU时,通常在安装配置GPU的环境过程中遇到问题:比如:安装TensorFlow2.1过程中,想要使用到电脑的显卡来进行开发,但是发现默认需要CUDATOOLKIT=10.1 ...
了解NVIDIA显卡驱动（包括CUDA、CUDA Driver、CUDA Toolkit、CUDNN、NCVV）
背景开发过程中需要用到GPU时,通常在安装配置GPU的环境过程中遇到问题:CUDA Toolkit和CUDNN版本的对应关系:CUDA和电脑显卡驱动的版本的对应关系:CUDA Toolkit.CUD ...
了解NVIDAI显卡驱动（包括：CUDA、CUDA Driver、CUDA Toolkit、CUDNN、NCVV）
转载一篇背景开发过程中需要用到GPU时,通常在安装配置GPU的环境过程中遇到问题:CUDA Toolkit和CUDNN版本的对应关系:CUDA和电脑显卡驱动的版本的对应关系:CUDA Toolk ...
ubuntu22.04安装显卡驱动+cuda+cudnn
ubuntu22.04安装显卡驱动+cuda+cudnn 1. 下载驱动和卸载.禁用自带驱动程序 1.1 查看系统显卡型号 1.2 从NVIDIA官网下载相应驱动 1.3 卸载Ubuntu自带的驱动程 ...

关于CUDA，cuDNN，TF，CUDA驱动版本兼容问题

基本概念

工作当中改如何做？

关于CUDA，cuDNN，TF，CUDA驱动版本兼容问题相关推荐

最新文章

热门文章