CUDA安装踩坑指南-Effective DeepLearning

最新推荐文章于 2025-07-12 22:33:29 发布

耳总21

最新推荐文章于 2025-07-12 22:33:29 发布

阅读量1.1w

点赞数 2

CC 4.0 BY-SA版权

分类专栏： tensorflow 文章标签：人工智能AI tensorflow cuda

本文链接：https://blog.youkuaiyun.com/lzcong1986/article/details/81050639

本文档详细记录了在Ubuntu 16.04和CentOS 7上安装CUDA的过程，特别强调了与TensorFlow集成时的注意事项。首先，选择合适的操作系统和CUDA版本至关重要，因为TensorFlow仅支持特定CUDA版本。然后，按照NVIDIA提供的步骤安装CUDA，但避免使用CUDA自带的显卡驱动，推荐从GeForce官网下载。安装过程中需注意环境变量设置，确保CUDA Toolkit的路径添加到PATH和LD_LIBRARY_PATH中。此外，解决可能出现的缺少依赖、版本不匹配、设备找不到等问题，如libcudnn的安装和显卡驱动兼容性问题。最后提醒不要使用apt-get自动更新显卡驱动。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

写在最前面, 因为我们用的tensorflow这个谷歌的AI框架，而且用的是官网的按照保，所以下面的都是按照tensorflow官网的要求和规则来做。其他自己框架或者自己编译的安装的小伙伴们不一定适用。
第一步其实并不是安装安装 NVIDIA cuda，而是先要选择好操作系统，这个很重要因为现在tensorflow官网只对ubuntu, maxOS, windows的错误进行维护，其他系统的错误就要靠自己囖。

说操作系统重要还有一点就是cuda会依赖不同的操作系统的自带驱动包，然后要根据你的显卡核心版本去下载不同版本的cuda的安装包，不同的tensorflow版本又只能支持对应版本的cuda驱动，如果错了一点可能又要删除重新弄，建议如果是新机器还是用最新的稳定版的系统为好。下面的步骤是针对ubuntu16.0.4和centOS 7的，而且亲测有效（花了两三天时间才装好的）

==========================================真正的开始===========================================

安装 NVIDIA cuda 

文档：

总目录：https://developer.nvidia.com/cuda-toolkit-archive 

最新版本：https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#runfile-nouveau 

自选版本：https://docs.nvidia.com/cuda/archive/9.0/cuda-installation-guide-linux/index.html  

1. 选择 runfile下载 cuda_9.0.176_384.81_linux-run, 在发帖之前tensorflow到了1.9版本，但是官方的安装包都只支持cuda9.0,更高版本的cuda需要自行修改源码去支持了，所以这里选择了9.0的包。在下载之前官网文档上还有一个Pre-installation Actions 的步骤，这个步骤里面的每一小步也是很重要的所以大