ubuntu16.04下安装tensorflow-gpu

最新推荐文章于 2025-03-31 21:32:50 发布

q012

最新推荐文章于 2025-03-31 21:32:50 发布

阅读量408

点赞数

分类专栏：人工智能文章标签： tensorflow ubuntu gpu python cuda

本文链接：https://blog.youkuaiyun.com/qq_43571525/article/details/108835810

版权

人工智能专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍了在Ubuntu16.04系统中安装TensorFlow-GPU的过程，包括选择合适的CUDA版本、下载cuDNN，以及确保TensorFlow与CUDA版本的兼容性。特别注意的是，Python版本的选择也会影响TensorFlow的安装。

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

ubuntu16.04下安装tensorflow-gpu
参考链接https://blog.youkuaiyun.com/u014595019/article/details/53732015
这里有几处需要注意：
1.安装CUDA的版本，终端输入 ubuntu-drivers devices 查看本系统对应的cuda推荐版本
参考链接：【ubuntu下查看cuda ，cudnn版本，以及对应tensorflow安装版本_qq_40212975的博客-优快云博客】https://blog.youkuaiyun.com/qq_40212975/article/details/89245685
上述链接也有下载cudnn的链接

2.安装tensorflow的链接：
tensorflow-gpu
tensorflow-cpu

3.注意tensorflow与CUDA的版本对应（很重要，不然会报错）：
tensorflow与CUDA版本对应关系
.whl文件下载地址
注意：这里的35/38指的就是python的版本，我这里的CDUA用的10.1，对应tensorflow选的1.14，但是tensorflow1.14不支持python3.8（包含）及以上的版本，所以就用ubuntu16.04自带的python3.5就好，千万不要卸掉。