Ubuntu16.04.5 配置英伟达NVIDIA 显卡 驱动实现GPU加速

Ubuntu16.04.5 配置英伟达NVIDIA 显卡 驱动实现GPU加速

标签(空格分隔): 运维系列


  • 一:系统环境初始化与系统包准备
  • 二:安装测试步骤

一:系统环境初始化与系统包准备

apt-get update 
apt-get install vim openssh-server

准备系统所需要的安装包

NVIDIA-Linux-x86_64-440.44.run

cuda_10.2.89_440.33.01_linux.run 

图片.png-20.9kB

二:安装测试步骤

1.1 安装Nvidia显卡驱动

1. 到官网上下载自己GPU对应版本的显卡驱动。

下载地址:https://www.nvidia.cn/Download/index.aspx?lang=cn

选择你的显卡驱动版本 点击搜索下载即可

图片.png-37.1kB

图片.png-23.4kB


1.2 安装NVIDIA-Linux-x86_64-440.44.run

屏蔽自带的显卡驱动

1) vim /etc/modprobe.d/blacklist.conf

2) 在最后一行加上:blacklist nouveau  ,这里是将Ubuntu自带的显卡驱动加入黑名单

3) 在终端输入:update-initramfs –u,使修改生效

4 ) 从新启动系统: reboot 

5)打开终端输入lsmod | grep nouveau,没有输出,则屏蔽成功

6 ) service lightdm stop 

图片.png-13.3kB

图片.png-5.4kB

图片.png-3.9kB

图片.png-3.7kB

安装 NVIDIA-Linux-x86_64-440.44.run

./NVIDIA-Linux-x86_64-440.44.run

图片.png-11.9kB

图片.png-5.7kB

图片.png-8.9kB

图片.png-19.9kB

图片.png-15.4kB

1.3 安装cuda_10.2.89_440.33.01_linux.run

1. 下载CUDA

下载地址:https://developer.nvidia.com/cuda-downloads

cuda_10.2.89_440.33.01_linux.run 

./cuda_10.2.89_440.33.01_linux.run 

图片.png-54.5kB

图片.png-14kB

图片.png-11.4kB

图片.png-18.9kB

配置环境变量

vim /etc/profile

----
到最后加上

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$PATH
----
source /etc/profile

图片.png-3kB

图片.png-36.1kB

测试是否安装成功

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

图片.png-40.2kB

图片.png-42kB

1.4 安装CuDNN


1. 下载

网址:https://developer.nvidia.com/rdp/cudnn-download

需要自己注册用户名与密码登录 才能下载
cudnn-10.2-linux-x64-v7.6.5.32.tgz
测试所需包

图片.png-45.1kB

 tar -zxvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

cd cuda/
cp include/cudnn.h /usr/local/cuda/include/
cp lib64/lib* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

图片.png-5.3kB

图片.png-21.9kB

验证是否安装成功

网址:https://developer.nvidia.com/rdp/cudnn-download
下载
libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb

图片.png-35.6kB


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 ldconfig 

图片.png-54.7kB


dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb

dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb

dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb

图片.png-32.6kB

图片.png-10kB

cp -r /usr/src/cudnn_samples_v7/ /home/el/
cd /home/el/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN

图片.png-34.7kB

图片.png-40.2kB

图片.png-30.6kB


1.5 安装anaconda3 

Anaconda3-2019.10-Linux-x86_64.sh 

chmod +x Anaconda3-2019.10-Linux-x86_64.sh 

vim /etc/profile

------
增加

export PATH=/opt/anaconda3/bin:$PATH
------

conda -V 

图片.png-22.7kB

图片.png-5kB

图片.png-6.9kB

1.6 安装opencv

1. 下载

网址:https://pypi.org/project/opencv-python/#files

因为安装的python是3.7的,所以opencv名字中要是"cp37"的。
想要安装opencv3,所以名字中要为opencv_python-3.****
我的系统是linux 64位的的,所以名字要是***linux1_x86_64**

软件:
opencv_python-4.1.2.30-cp37-cp37m-manylinux1_x86_64.whl

pip install opencv_python-4.1.2.30-cp37-cp37m-manylinux1_x86_64.whl

conda list |grep opencv

图片.png-11kB

图片.png-3.7kB

1.7更新系统cmake版本

apt-get install cmake
cmake --version 

图片.png-35.9kB

图片.png-4.1kB

在Ubuntu16.04默认安装的cmake版本为3.5.x,可通过一下命令,查看版本。

cmake --version

有时需要安装高版本的cmake。

1.卸载旧版本

apt-get autoremove cmake

2.以安装3.12.3版本为例

$ sudo apt-get install build-essential
$ wget http://www.cmake.org/files/v3.12/cmake-3.12.3.tar.gz

3.解压、安装
$ tar xf cmake-3.12.3.tar.gz
$ cd cmake-3.11.3
$ ./configure
$ make
$ sudo make install

4.解决路径问题

export PATH=/usr/local/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
cmake

5.查看,安装成功

cmake --version

图片.png-6.5kB

1.8 配置xgboost 支持

1. 下载源代码

apt-get install git

git clone --recursive https://github.com/dmlc/xgboost

2. 编译GPU共享库

cd xgboost
mkdir build
cd build
cmake .. -DUSE_CUDA=ON
make -j

3. 安装Python包
在xgboost根目录下

cd python-package
sudo python3 setup.py install

测试GPU加速

python3 tests/benchmark/benchmark.py

图片.png-34.9kB

图片.png-41.3kB

图片.png-14.8kB

猜你喜欢

转载自blog.51cto.com/flyfish225/2461788
今日推荐