[关闭]
@zhangyy 2020-01-16T14:06:41.000000Z 字数 3030 阅读 145

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

运维系列


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

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

  1. apt-get update
  2. apt-get install vim openssh-server

  1. 准备系统所需要的安装包
  2. NVIDIA-Linux-x86_64-440.44.run
  3. cuda_10.2.89_440.33.01_linux.run

图片.png-20.9kB

二:安装测试步骤

1.1 安装Nvidia显卡驱动

  1. 1. 到官网上下载自己GPU对应版本的显卡驱动。
  2. 下载地址:https://www.nvidia.cn/Download/index.aspx?lang=cn
  3. 选择你的显卡驱动版本 点击搜索下载即可

图片.png-37.1kB

图片.png-23.4kB


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

  1. 屏蔽自带的显卡驱动
  2. 1 vim /etc/modprobe.d/blacklist.conf
  3. 2 在最后一行加上:blacklist nouveau ,这里是将Ubuntu自带的显卡驱动加入黑名单
  4. 3 在终端输入:update-initramfs u,使修改生效
  5. 4 ) 从新启动系统: reboot 
  6. 5)打开终端输入lsmod | grep nouveau,没有输出,则屏蔽成功
  7. 6 ) service lightdm stop

图片.png-13.3kB

图片.png-5.4kB

图片.png-3.9kB

图片.png-3.7kB

  1. 安装 NVIDIA-Linux-x86_64-440.44.run
  2. ./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. 1. 下载CUDA
  2. 下载地址:https://developer.nvidia.com/cuda-downloads
  3. cuda_10.2.89_440.33.01_linux.run
  4. ./cuda_10.2.89_440.33.01_linux.run

图片.png-54.5kB

图片.png-14kB

图片.png-11.4kB

图片.png-18.9kB

  1. 配置环境变量
  2. vim /etc/profile
  3. ----
  4. 到最后加上
  5. export PATH=/usr/local/cuda/bin:$PATH
  6. export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$PATH
  7. ----
  8. source /etc/profile

图片.png-3kB

图片.png-36.1kB

  1. 测试是否安装成功
  2. cd /usr/local/cuda/samples/1_Utilities/deviceQuery
  3. make
  4. ./deviceQuery

图片.png-40.2kB

图片.png-42kB

1.4 安装CuDNN

  1. 1. 下载
  2. 网址:https://developer.nvidia.com/rdp/cudnn-download
  3. 需要自己注册用户名与密码登录 才能下载
  4. cudnn-10.2-linux-x64-v7.6.5.32.tgz
  5. 测试所需包

图片.png-45.1kB

  1. tar -zxvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
  2. cd cuda/
  3. cp include/cudnn.h /usr/local/cuda/include/
  4. cp lib64/lib* /usr/local/cuda/lib64/
  5. sudo chmod a+r /usr/local/cuda/include/cudnn.h
  6. sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

图片.png-5.3kB

图片.png-21.9kB

  1. 验证是否安装成功
  2. 网址:https://developer.nvidia.com/rdp/cudnn-download
  3. 下载
  4. libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
  5. libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb

图片.png-35.6kB


  1. cd /usr/local/cuda/lib64/
  2. sudo rm -rf libcudnn.so libcudnn.so.7
  3. sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7
  4. sudo ln -s libcudnn.so.7 libcudnn.so
  5. sudo ldconfig

图片.png-54.7kB

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

图片.png-32.6kB

图片.png-10kB

  1. cp -r /usr/src/cudnn_samples_v7/ /home/el/
  2. cd /home/el/cudnn_samples_v7/mnistCUDNN
  3. make clean && make
  4. ./mnistCUDNN

图片.png-34.7kB

图片.png-40.2kB

图片.png-30.6kB


1.5 安装anaconda3 

  1. Anaconda3-2019.10-Linux-x86_64.sh
  2. chmod +x Anaconda3-2019.10-Linux-x86_64.sh
  3. vim /etc/profile
  4. ------
  5. 增加
  6. export PATH=/opt/anaconda3/bin:$PATH
  7. ------
  8. conda -V

图片.png-22.7kB

图片.png-5kB

图片.png-6.9kB

1.6 安装opencv

  1. 1. 下载
  2. 网址:https://pypi.org/project/opencv-python/#files
  3. 因为安装的python3.7的,所以opencv名字中要是"cp37"的。
  4. 想要安装opencv3,所以名字中要为opencv_python-3.****
  5. 我的系统是linux 64位的的,所以名字要是***linux1_x86_64**
  6. 软件:
  7. opencv_python-4.1.2.30-cp37-cp37m-manylinux1_x86_64.whl
  8. pip install opencv_python-4.1.2.30-cp37-cp37m-manylinux1_x86_64.whl
  9. conda list |grep opencv

图片.png-11kB

图片.png-3.7kB

1.7更新系统cmake版本

  1. apt-get install cmake
  2. cmake --version

图片.png-35.9kB

图片.png-4.1kB

  1. Ubuntu16.04默认安装的cmake版本为3.5.x,可通过一下命令,查看版本。
  2. cmake --version
  3. 有时需要安装高版本的cmake
  4. 1.卸载旧版本
  5. apt-get autoremove cmake
  6. 2.以安装3.12.3版本为例
  7. $ sudo apt-get install build-essential
  8. $ wget http://www.cmake.org/files/v3.12/cmake-3.12.3.tar.gz
  9. 3.解压、安装
  10. $ tar xf cmake-3.12.3.tar.gz
  11. $ cd cmake-3.11.3
  12. $ ./configure
  13. $ make
  14. $ sudo make install
  15. 4.解决路径问题
  16. export PATH=/usr/local/bin:$PATH
  17. export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
  18. cmake
  19. 5.查看,安装成功
  20. cmake --version

图片.png-6.5kB

1.8 配置xgboost 支持

  1. 1. 下载源代码
  2. apt-get install git
  3. git clone --recursive https://github.com/dmlc/xgboost
  4. 2. 编译GPU共享库
  5. cd xgboost
  6. mkdir build
  7. cd build
  8. cmake .. -DUSE_CUDA=ON
  9. make -j
  10. 3. 安装Python
  11. xgboost根目录下
  12. cd python-package
  13. sudo python3 setup.py install
  14. 测试GPU加速
  15. python3 tests/benchmark/benchmark.py

图片.png-34.9kB

图片.png-41.3kB

图片.png-14.8kB

添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注