记一次在ubuntu16.04上安装tensorflow-gpu的过程

本文详细记录了在Ubuntu 16.04系统上安装tensorflow-gpu 1.4.1及CUDA 8.0的过程,包括遇到的坑与解决方案。由于阿里云pypi镜像只提供tensorflow-gpu 1.4.1,故选择该版本,硬件环境为NVIDIA Tesla M40显卡。遵循的主要步骤包括安装CUDA,cudnn,最后是tensorflow-gpu,过程中需特别注意避免安装错误的CUDA版本。

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

前言:
tensorflow已经安装了很多遍了,也填平了好多坑了。XX说过:“世界上本来没有大牛,爬的坑多了就有了大牛”。

这是一篇安装类的文章,这类文章的时效性非常高。注意我安装的时间是2018-01-05。这个时候这个世界的snapshot是:ubuntu17.04, tensorflow-gpu 1.5.0, cuda9.1, cudnn7.1都已经有了。但是我这次安装是在阿里云上安装的,在阿里云的pypi镜像上只有1.4.1的所以我也只安装1.4.1的。(pip配置http://mirrors.aliyun.com/pypi/simple/tensorflow-gpu/)

因此我本次安装的是:ubuntu16.04 + tensorflow1.4.1 + cuda8.0 + cudnn6.0 .
硬件是 NVIDIA Corporation GM200GL [Tesla M40] (rev a1)。 阿里云上最便宜的gn4。

主要思路是参考:https://tensorflow.google.cn/install/install_linux。无非三部:一,cuda,二cudnn,三tensorflow

1 安装cuda8.0

大而全的无用的完全安装手册: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions
这个手册太复
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值