一、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

(出现下图即安装成功)

vGPU兼容平台 vgpu 安装_cuda

二、 Cuda8安装

下表是nvidia与cuda的版本对应表

vGPU兼容平台 vgpu 安装_linux_02

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

后面的数字是指定版本号,可不加