1.close nouveau
终端输入:sudo gedit /etc/modprobe.d/blacklist.conf
末尾加两行
blacklist nouveau
options nouveau modeset=0
然后保存
之后运行这行命令
sudo update-initramfs -u
重启,然后运行:
lsmod | grep nouveau
结果应该不会显示任何东西
2.Install NVIDIA Driver
查看显卡型号和推荐的显卡驱动(显卡驱动的版本直接就决定了cuda的版本,而cuda版本也就决定了cudnn和tensorflow的版本,所以显卡驱动务必安装正确)
ubuntu-drivers devices
安装驱动,方式一(取决于下载速度,通常会很慢):
# 将系统中存在的nvidia驱动全部卸载掉(如果有的话)
sudo apt-get purge nvidia*
# 添加ppa源并更新
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
# 安装nvidia-390版本
sudo apt-get install nvidia-390
方式二:打开 Software & Updates ,点击”Additional Drivers” ,选择第一个选项Using NVIDIA driver metapackage from nvidia-driver-390(这种方式 取决于下载速度,通常会很慢)
方式三:从官网下载驱动文件NVIDIA-Linux-x86_64-390.116.run,然后手动安装(主要安装驱动需要gcc-7,和make)
sudo chmod +x
NVIDIA-Linux-x86_64-390.116.run
sudo ./
NVIDIA-Linux-x86_64-390.116.run
-no-nouveau-check -no-opengl-files
-no-nouveau-check
:安装驱动时禁用
nouveau
-no-opengl-files:只安装驱动文件,不安装
OpenGL
文件
测试是否安装成功
nvidia-smi
3.Install CUDA9.0
下载run file (cuda_9.0.176_384.81_linux.run)
sudo ./cuda_9.1.85_387.26_linux.run –override
注意:
-
--override is necessary.
-
When installing, it will ask you whether install driver either. Answer NO because you have install it earlier!
4.Install cudnn7
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
5. Install tensorflow
由于安装的是cuda9.0,选择tensorflow_gpu-1.5.0-cp36-cp36m-linux_x86_64.whl,最新版本的tensorflow-gpu与cuda9.0不兼容
下载地址:https://www.anaconda.com/download/
根据系统选择对应的安装包(
下载完成后,cd到存放目录并执行:
bash Anaconda3-xxxx.sh #xxxx此处对应自己的包
按照提示安装即可,安装路径可以选择默认。
安装完之后执行:
source ~/.bashrc
检查是否安装成功
终端执行:conda list
若提示命令无法找到,则修改.profile文件:
sudo gedit ~/.profile
添加以下几行:
if [ -d "$HOME/anaconda3/bin" ] ; then
PATH="$HOME/anaconda3/bin:$PATH"
fi
使生效:
source
~
/
.profile
conda create -n tensorflow pip python=3.6 #python
版本根据要安装的
tensorflow
选择
激活环境:
source activate tensorflow
安装tensorflow:
pip install tensorflow_gpu-1.5.0-cp36-cp36m-linux_x86_64.whl
安装过程中会下载其他的依赖包,下载可能会很慢,可以先把相关依赖包安装完,再安装tensorflow。
验证安装
终端输入:python
然后输入以下代码:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))