ubuntu18+gtx1060 +cuda9+cudnn-v7+opencv3.1.0 配置深度学习环境

时间:2023-12-11 19:15:32

将笔记本的ubuntu系统更新到18版本后重新配置深度学习环境,在此记载方便日后参考

具体配置为 Ubuntu18.04+gtx1060+opencv-3.1

第1步 安装依赖包

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

sudo apt-get install --no-install-recommends libboost-all-dev

sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

sudo apt-get install git cmake build-essential

要确保这些依赖包都成功安装,不然后续出错让人忧伤

第2步 禁用nouveau

禁用nouveau后才能顺利安装NVIDIA显卡驱动
先通过命令打开配置文件

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

然后写入

blacklist nouveau option nouveau modeset=0 

保存文件后关闭,更新后才能生效

sudo update-initramfs -u

第3步 配置环境变量

打开配置文件

sudo gedit ~/.bashrc

写入

export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH 

保存后退出

第4步 安装测试cuda(默认已安装好显卡驱动)

官网 https://developer.nvidia.com/cuda-downloads
下载后缀为.run 的文件
执行如下命令安装

sudo sh cuda_9.0.176_384.81_linux.run

回车键可加速翻页
其中出现如下选项,不知是否会出错,有点忧伤
You are attempting to install on an unsupported configuration. Do you wish to continue?

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
(y)es/(n)o/(q)uit: no
这一选项要选择no

然后进行测试

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

出现如下信息则安装成功

deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 9.1, CUDA Runtime Version = 9.0, NumDevs = 1
Result = PASS

配置环境变量

sudo gedit /etc/profile   ###修改计算机环境

或者

sudo gedit ~/.bashrc  ###修改用户环境

写入

export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH

重启电脑

sudo reboot  ###修改计算机环境后
source ~/.bashrc ###修改用户环境后

第5步 安装cudnn

官网:https://developer.nvidia.com/rdp/cudnn-download 下载cuda对应版本的cudnn
本人下载的是cudnn-9.1-linux-x64-v7.tgz
百度网盘链接链接: https://pan.baidu.com/s/1rPEoJEzAmg-3g6juZSSmmA 密码: vpx6
解压后得到cuda文件夹

cd include && sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件
cd .. && cd lib64
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7  #删除原有动态文件
sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7  #生成软衔接
sudo ln -s libcudnn.so.7 libcudnn.so  #生成软链接  

然后将路径添加到动态库

sudo gedit /etc/ld.so.conf.d/cuda.conf

写入

/usr/local/cuda/lib64  

保存后更新

sudo ldconfig 

安装完成后可用 nvcc -V 命令验证是否安装成功,若出现以下信息则表示安装成功:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176

第6步 安装opencv3.1

官网 : http://opencv.org/releases.html
下载合适版本,解压后进入opencv目录

mkdir build  && cd build    # 创建编译的文件目录
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..  

make -j8  #编译
sudo make install

安装完成后通过查看 opencv 版本验证是否安装成功:

pkg-config --modversion opencv  

卸载opencv的方法
进入 opencv/build 目录下

sudo make uninstall  

然后

sudo rm -r /usr/local/include/opencv2 /usr/local/include/opencv /usr/include/opencv /usr/include/opencv2 /usr/local/share/opencv /usr/local/share/OpenCV /usr/share/opencv /usr/share/OpenCV /usr/local/bin/opencv* /usr/local/lib/libopencv  

第7步安装tensorflow

pip install --index-url https://pypi.douban.com/simple tensorflow-gpu==1.8.0

第8步 安装pyTorch

先安装好 Anaconda
百度网盘
链接:https://pan.baidu.com/s/1TKmzVy8yY00RbCESCP2AHA 密码:hfjv

环境变量设置

sudo gedit  ~/.bashrc
$ export PATH=/home/ye/anaconda3/bin:$PATH

给conda增加源

 # 优先使用清华conda镜像
 conda config --prepend channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  # 也可选用科大conda镜像
 conda config --prepend channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
cat ~/.condarc #查看文件内容有无国内源信息,若有,则成功

安装 pytorch==0.4.0,torchvision

conda install pytorch==0.4.0 #cpu版本,可以根据需求改为gpu版本
conda install torchvision

测试,若

import torch

无报错,则成功