一、nvidia的安装
安装的方式有多种,这里主要采用.run文件安装
**
1
sudo apt-get purge nvidia*
卸载旧驱动,这一步通常不是必须的,安装nvidia时旧版本会自动覆盖
2
lsmod | grep nouveau
查询nouveau nvidia驱动,是否为启用状态,若看到nouveau对应的值为0,则跳过步骤3
3, 禁用自带的 nouveau nvidia驱动
禁用的驱动有两种方式,通常用第一种就可以解决,
3.1 方式1
1) 创建一个文件通过命令
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
2) 在创建的配置文件中输入
lacklist nouveau
options nouveau modeset=0
3)执行更新使配置文件生效
sudo update-initramfs -u
4)查看nouveau禁用情况,出现步骤一图中的效果即完成禁用
lsmod | grep nouveau
3.2方式2
1)进入ubuntu引导页时,点击键盘的‘e’键进入编辑grub启动项参数
2)找到$vt_handoff这个词,在前面添加nomodeset_,然后按F10启动,启动后 图标变大,不慌,继续进行下面的步骤
4、 禁用X-Window服务
sudo service lightdm stop
然后按Ctrl-Alt+F1进入命令行界面,输入用户名和密码登录即可
5、 给驱动run文件赋予执行权限:
sudo chmod +x NVIDIA-Linux-x86_64-390.87.run
(文件名具体相关的驱动名称)
6、 安装驱动
sudo ./NVIDIA-Linux-x86_64-390.87.run –no-opengl-files
(–no-opengl-files防止循环登录的问题)
7、检测nvidia是否安装成功
nvidia-smi
(出现下图即安装成功)
二、 Cuda8安装
下表是nvidia与cuda的版本对应表
1、关闭x_windows,安装cuda
sudo service lightdm stop
按Ctrl + Alt + F1 进入文本模式,
2、运行 CUDA 安装文件进行安装
sudo sh cuda_8.0.61_375.26_linux.run --no-opengl-libs
3、具体步骤
其中 cuda_8.0.61_375.26_linux.run 是我的 CUDA 安装文件名,而你需替换为自己的 CUDA 安装文件名
执行此命令约1分钟后会出现 0%信息,此时按空格键让此百分比增长,直到100%,然后按照提示操作即可,先输入 accept ,然后让选择是否安装 nvidia 驱动,这里的选择对应第2步开头,若未安装则输入 “y”,若确保已安装正确驱动则输入“n”。
剩下的选择则都输入“y”确认安装或确认默认路径安装,开始安装,此时若出现安装失败提示则可能为未关闭桌面服务或在已安装 nvidia 驱动的情况下重复再次安装 nvidia 驱动,安装完成后输入重启命令重启
4、配置cuda环境变量
1) 编辑.bashrc
sudo gedit ~/.bashrc
2) 在文末尾添加
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
3) 文件生效
source ~/.bashrc
4) 检查是否安装成功cuda
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
三、 Cudnn5.0安装
我们这里以版本为:cudnn-8.0-linux-x64-v5.1.tgz为例
1、 解压
tar -zxvf cudnn-8.0-linux-x64-v5.1.tgz
2、 复制头文件
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
3、 复制动态链接库
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
4、 删除原有动态文件
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5
5、 生成软链接
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so
6、 cuda安装验证
cd ~
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
四、 tensorflow-gpu离线安装
1、 准备以上的whl包
2、 在终端以pip3 install xxx.whl 方式安装
3、 安装顺序
Werkzeug
Numpy
Wheel
Bleach
backports.weakref
html5lib
markdown
protobuf
tensorflow-gpu
ps:
pip3安装只适合whl文件,若为gz压缩包,则是解压后用命令python3 setup.py install 安装,其余步骤一致
在线安装tensorflow-gpu只需将命令
pip3 install xxx.whl
改成
sudo pip install tensorflow-gpu==1.2.0
后面的数字是指定版本号,可不加