ubuntu16.04 安装tensorflow-1.5.0走过的坑

基本上我是跟着这个 博客 走的,但是也有一些问题,故写下这篇作为参考。

〇、我的环境

系统:Ubuntu16.04
显卡:gtx960
内存:8G

坑一、安装显卡驱动

这里有个坑,电脑的BIOS一定要关闭安全启动(securlty boot),至少我的是要关闭才能安装驱动成功。(有的说也要关闭快速启动,反正我都关了)

然后有两种常见方式安装驱动,都可以,ppa安装有点慢(看网速,我网慢),run文件安装方式相对来说复杂一些,都可以。

坑二、安装TensorFlowtensorflow-1.5.0

这里也有多种安装方式,但是目前pip安装只能支持cuda8.0版本,我安装了cuda9.1的版本,所以只能采用源码编译安装。

这里也有个坑,编译的时候错误,说找不到math_functions.hpp。

这是因为CUDA 9.1, math_functions.hpp 在cuda/include/crt/math_functions.hpp, 而不是在原来的位置 cuda/include/math_functions.hpp (CUDA 9.0 does)然后就报错了,只要重新建立软链接即可,执行命令:

sudo ln -s /usr/local/cuda/include/crt/math_functions.hpp /usr/local/cuda/include/math_functions.hpp

坑三、内存太小导致编译报错

创建交换分区解决这个问题

至此,成功完成安装。

四、参考网站

学习笔记(三)ubuntu16.04下Anaconda及tensorflowGPU版本的安装配置

Fatal error while compiling Tensorflow with CUDA 9.1 #15389

Ubuntu 16.04创建Swap分区或增加Swap分区容量

你可能感兴趣的:(ubuntu16.04 安装tensorflow-1.5.0走过的坑)