Ubuntu Anaconda安装TensorFlow-GPU
目前时间是2017.10.22
,目前的CUDA
最新版本是CUDA9
,TensorFlow
的版本是1.3
,一开始安装CUDA9
,后来发现TensorFlow
不支持,网上有人说可以通过源代码安装TensorFlow
,但是我觉得太麻烦了,后来就换了CUDA8
,中间遇到了很多版本不对应所产生的问题,下面详细列一下本次安装所使用的安装包的版本号:(亲测可安装成功)
- CUDA:
cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
- cudnn:
cudnn-8.0-linux-x64-v5.1.tar
- Anaconda:
Anaconda3-5.0.0.1-Linux-x86_64.sh
一、安装准备
CUDA下载
cudnn下载
Anaconda下载
在实际的安装过程中,发现目前最新的版本 Tensorflow 并不能支持,后来折腾了好久,才找到上面提到的那几个对应的安装包,百度网盘分享链接,密码:
8lpv
二、安装 NVIDIA 驱动
首先打开 terminal
,输入一下指令,更新应用列表
sudo apt-get update
多亏了Linux
社区的强大支持,我们可以用很简洁的方式安装驱动:
按下
win
按键,打开菜单然后在上方的
type to search
中输入:additional drivers
打开 "additional drivers —— 附加驱动
",然后选择 与自己显卡匹配的NVIDIA
驱动,我这里是using nvidia binary drivers (375)
然后选择 应用更改,这里安装可能会失败,此时你可以继续选择应用更改,多安装几次即可。
安装完成后,选择
restart
三、gcc 降版本
CUDA8.0
不支持gcc 5.0
及以上的编译器,而系统自带的是5.4
及以上版本,因此我们需要降版本,否则会在后面报错。
在terminal
中输入如下指令,将gcc
版本降到 4.9
sudo apt-get install g++-4.9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives --set c++ /usr/bin/g++
四、安装 CUDA 8.0
进入cuda
安装包所在文件夹,右键打开终端,在终端中输入以下代码
# 根据官网上的提示安装 cuda 8.0
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
五、安装 cudnn
把cudnn-8.0-linux-x64-v5.1.tar
解压,右键打开终端,在终端中输入以下代码
sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 复制到 include 中
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 # 复制到 lib64 中
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* # 将头文件复制进去
六、配置CUDA环境变量
开启 GPU 支持:
根据官网教程,我们在terminal
中键入下列命令:
sudo gedit ~/.bash_profile # 打开.bash_profile 这是用户的环境变量,不是全局的
然后在打开的文本末尾加入:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
保存并关闭后,输入下列命令使环境变量生效:
source ~/.bash_profile # 使被更改的环境变量生效
安装完成后,一定要通过如下两条命令对驱动进行检验:
-
NVIDIA
的设置界面nvidia-settings # 打开 NVIDIA 设置界面
-
NVIDIA GPU
列表nvidia-smi # 这条指令会在 terminal 中产生GPU列表
网上有些人抄别人的博客,说
nvcc -V
就可以验证,经过大神实测,存在nvcc -V
正常输出但是驱动仍安装失败的现象,因此,上述验证方法是不可信的。
七、安装Anaconda
进入anaconda
安装包所在目录,打开终端,根据版本输入安装命令:
bash Anaconda3-5.0.0.1-Linux-x86_64.sh
回车后,是许可文件,接收许可,anaconda
将安装在~/anaconda
目录下,直接回车就好。
最后会询问是否把anaconda
的bin
添加到用户的环境变量中,选择yes
。
安装成功以后,在终端输入
source ~/.bashrc
让.bashrc
中添加的路径生效。
当然,你也可以手动完成环境变量的修改:
-
首先打开环境变量的文件
gedit ~/.bashrc
-
然后在文件末尾加入 anaconda3 的路径
export PATH=/home/你的路径/anaconda3/bin:$PATH
-
最后使我们的改动生效
source ~/.bashrc
这样,我们在terminal
中输入python
就会默认打开anaconda3
里默认的python
了
八、安装Tensorflow-GPU
创建Tensorflow
的运行环境
conda create -n tensorflow python=3.5
激活环境,使用 pip 安装 TensorFlow-GPU
source activate tensorflow
pip install tensorflow-gpu keras # 安装 gpu 版本的 tensorflow 和 keras
安装完成后,我们使用如下命令,即可检验是否成功:
python -c "import keras"
如果看到如下输出,就说明安装成功
参考文章
Ubuntu 安装 tensorflow-gpu + keras
Ubuntu17.04+Nvidia GTX 1080+CUDA 9.0+cuDNN 7.0+TensorFlow 1.3
CUDA 9.0 + cuDNN 7.0 + Tensorflow源码编译