Ubuntu 16.04 + 显卡驱动 + Cuda9.1 + Cudnn7.05 + Opencv3.4 + Caffe2 环境搭建

1.显卡驱动安装：
禁用系统自带的独立显卡的驱动nouveau：
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
写入：
blacklist nouveau
options nouveau modeset=0
然后更新一下：
sudo update-initramfs -u
修改后需要重启系统。确认下Nouveau是已经被你禁用，使用命令：
lsmod | grep nouveau
2.ctrl+alt+f1 进入显卡驱动所在文件路径：
sudo service lightdm stop
sudo sh NVIDIA-Linux-x86_64-340.102.run
(弹出选择窗口默认都选接受和yes)
sudo service lightdm start
cat /proc/driver/nvidia/version 会输出NVIDIA Driver的版本号

注意：如果以后遇到因显卡驱动无法进入系统的情况直接使用步骤2重装驱动覆盖之

3.Cuda9.1安装：
ctrl+alt+f1 进入cuda.run所在路径
sudo service lightdm stop
sudo sh cuda_9.1.85_387.26_linux
注意：提示是否安装图形驱动时选择no其他选择yes或默认
sudo service lightdm start
ctrl+alt+f7
4.设置环境变量终端中输入:
sudo gedit /etc/profile
在打开的文件末尾，添加以下两行:
export PATH=/usr/local/cuda-9.1/bin: $PATH export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64:$ LD_LIBRARY_PATH
然后输入:
source /etc/profile
5.重启电脑检查上述的环境变量是否设置成功终端输入：
nvcc -V
会输出Cuda的版本信息

6.cuDNN7.05安装： Download
https://developer.nvidia.com/rdp/cudnn-download
Ubuntu 16.04 + 显卡驱动 + Cuda9.1 + Cudnn7.05 + Opencv3.4 + Caffe2 环境搭建

7.进入下载好的两个文件路径：
sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.1_amd64.deb
sudo dpkg -i libcudnn7-dev_7.0.5.15-1+cuda9.1_amd64.deb
注意先后顺序

8.Opencv3.4.0安装：
安装依赖库和cmake:
sudo apt-get update
sudo apt-get install cmake
sudo apt-get install build-essential libgtk2.0-dev libavcodec-dev libavformat-dev libjpeg.dev libtiff4.dev libswscale-dev libjasper-dev

9.创建编译文件夹并编译安装：
mkdir my_build_dir
cd my_build_dir
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local …
sudo make
sudo make install

10.Caffe2安装：
安装依赖库：
sudo apt-get update
sudo apt-get install -y --no-install-recommends
build-essential
cmake
git
libgoogle-glog-dev
libgtest-dev
libiomp-dev
libleveldb-dev
liblmdb-dev
libopenmpi-dev
libsnappy-dev
libprotobuf-dev
openmpi-bin
openmpi-doc
protobuf-compiler
python-dev
python-pip
python-setuptools
libgflags-dev
libeigen3-dev
sudo pip install
wheel
future
numpy
protobuf
hypothesis
11.创建根目录：
mdkir ~/bin && cd ~/bin
12.github获取源码下载并编译：
git clone–recursive https://github.com/caffe2/caffe2.git && cd caffe2
make && cd build && sudo make install
13.在环境中配置如下命令：

	export PYTHONPATH=/usr/local:$PYTHONPATH
	export PYTHONPATH=$PYTHONPATH:$HOME/bin/caffe2/build
	export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
	export CMAKE_LIBRARY_PATH=$CMAKE_LIBRARY_PATH:$HOME/bin/caffe2/third_party/nccl/build/lib

测试是否安装成功：

cd ~ && python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

如果显示Success则大功告成

Ubuntu 16.04 + 显卡驱动 + Cuda9.1 + Cudnn7.05 + Opencv3.4 + Caffe2 环境搭建

相关推荐