ubuntu16.04安装tensorflow1.9(gpu)版本所需软件:nvidia-415,cuda9.0,cudnn7.0,anaconda3-4.4.0
软件下载:
(1)cuda9.0下载地址:
https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal
ps:后续cuda的安装步骤按照官网下载地址页面的代码进行操作
(2)cudnn7.0下载地址:https://developer.nvidia.com/cudnn
进入网址之后需要注册账号,然后才能下载
(3)anaconda3-4.4.0下载地址:https://www.anaconda.com/download/
选择python3.6版本。若官网版本已更新,则在网页搜索
一.安装nvidia显卡驱动
1.卸载系统里的Nvidia低版本显卡驱动
sudo apt-get purge nvidia*
2.把显卡驱动加入PPA
sudo -E add-apt-repository ppa:graphics-drivers
sudo apt-get update
3.查找显卡驱动的版本并安装
使用终端命令查看Ubuntu推荐的驱动版本:
ubuntu-drivers devices
(本机是nvidia-415)
采用apt-get命令在终端安装:
sudo apt-get install nvidia-415 nvidia-settings nvidia-prime
4.重启系统并验证
在终端输入以下命令行:
方法1:
lsmod | grep nvidia
如果没有输出,则安装失败。
方法2:查看nvidia版本
nvidia-smi
输入命令行,查看Ubuntu自带的nouveau驱动是否运行:
lsmod | grep nouveau
如果终端没有内容输出,则显卡驱动的安装成功
二 安装cuda 9.0
1.在安装包路径打开终端,输入以下命令(在前面cuda9.0下载地址页面有显示操作代码):
sudo dpkg-i-cuda-repo-ubuntu-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
2.配置环境变量
运行如下命令打开profile文件
sudo gedit /etc/profile
打开文件后在文件末尾添加路径,也就是安装目录(根据自己的路径),命令如下:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
保存,然后重启电脑
sudo reboot
3.验证cuda是否安装成功
方法1:
输入以下命令:
nvcc –V
或者 nvcc --version
方法2:查看cuda版本
cat /usr/local/cuda/version.txt
三、安装cudnn
安装cudnn比较简单,就是复制几个文件:库文件和头文件。将cudnn的头文件复制到cuda安装路径的include路径下,将cudnn的库文件复制到cuda安装路径的lib64路径下。
1.在安装包路径打开终端,执行下面的指令安装:
tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
cd cuda
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
然后更新网络连接:
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.7.0.5 //自己查看.so的版本对应更改
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
2.检验cudnn是否安装成功
方法1:
nvcc –V
方法2:查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
四 安装anaconda
1 使用命令:
bash Anaconda3-4.4.0-Linux-x86_64.sh
2 验证是否安装成功
方法1:
anaconda –V
方法2:查看conda版本
conda –V
五 安装TensorFlow
1 输入以下命令安装TensorFlow1.9
conda create -n tensorflow
source activate tensorflow
pip install --ignore-installed --upgrade https://download.tensorflow.google.cn/linux/gpu/tensorflow_gpu-1.9.0-cp36-cp36m-linux_x86_64.whl
安装完成后,输入以下指令关闭TensorFlow:
source deactivate
2 验证是否安装成功
方法1:
终端输入python,进入python命令行后输入:
import tensorflow as tf
没有报错表示安装成功
方法2:查看TensorFlow版本
输入以下指令:
python
import tensorflow as tf
tf.__version__