欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

caffe ubuntu14 gpu环境安装

程序员文章站 2022-05-09 22:56:51
...

1.安装依赖

sudo apt-get install build-essential cmake git
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

#boost库我源码安装的 boost_1_61_0.zip [参考](https://blog.csdn.net/tanglj86/article/details/79304918)
进入目录./bootstrap.sh
./b2
sudo ./b2 install

2.禁用nonveau驱动(没有就跳过)

打开终端,

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

在最后添加
blacklist nouveau
options nouveau modset=0

保存退出后执行
sudo update-initramfs -u

禁用结束。重启电脑后

sudo lspci | grep nouveau

如果没有内容,则禁用成功。

(下面是ubuntu1604的case)

sudo gedit /etc/modprobe.d/blacklist.conf

在最后添加
blacklist rivafb
blacklist vga16fb
blacklist nouveau
blacklist nvidiafb
blacklist rivatv

完成后
sudo update-initramfs -u

3.安装nv驱动

去nv官网 查找gt720的linux驱动,比如我选择 NVIDIA-Linux-x86_64-390.77.run(GT1030也是安装的这个驱动, 不推荐安装太新的显卡驱动) .
安装的时候需要 ctrl+alt+f1 进入非gui界面,然后sudo init 3 (这会停掉 x 服务),sudo service lightdm stop,sudo ./NVIDIA-Linux-x86_64-390.77.run --no-x-check --no-nouveau-check --no-opengl-files
安装成功的话, 重启后 #nvidia-smi 会显示显卡信息

4.安装cuda库和cudnn,这里就不要在安装nvidia driver了,直接安装toolkit

cuda_8.0.44_linux.run + cudnn-8.0-linux-x64-v7.1.tgz, 可以直接参考 https://www.linuxidc.com/Linux/2017-01/139300.htm

 

ref  https://blog.csdn.net/ksws0292756/article/details/79160742