基本上我是跟着这个 博客 走的,但是也有一些问题,故写下这篇作为参考。
〇、我的环境
系统: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版本的安装配置