ubuntu 安装 nvidia 环境

本文共有2563个字,关键词:nvidiaCUDAcuDNN

事前准备

确认电脑没有安装 nvidia 驱动,如果有需要先卸载

dpkg -l | grep nvidia
dpkg -l | grep cuda

卸载方法,卸载后需要重现启动

sudo apt-get remove nvidia*
sudo apt-get remove cuda*
sudo apt autoremove

开始安装显卡驱动

Ctrl+Alt+F1进入 CUI 纯命令界面

如需恢复 GUI 界面按ALT + F7

停止 X Server

sudo service lightdm stop
pkill Xorg

软件源添加 ppa 的地址

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

安装nvidia-415 (这里415可以换成最新稳定版本,430安装后不能启动,原因未知)

sudo apt-get install nvidia-415
reboot #安装完成后需要重启

确认显卡驱动安装成功

dmesg | grep nvidia
dpkg -l | grep nvidia

如果出现ii nvidia-390 390.48-0ubuntu0~gpu16.04.3则说明安装成功
输入nvidia-smi也可查看详情

安装 CUDA

从NVIDIA 官网下载 安装文件https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=debnetwork

根据网页提示运行

sudo dpkg -i cuda-repo-ubuntu1604_10.1.168-1_amd64.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

把路径信息加到~/.bashrc文件的最后

## CUDA and cuDNN paths
export PATH=/usr/local/cuda/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}

reboot重启电脑后测试工作是否正常

echo $PATH             # 输出是否包含"/usr/local/cuda/bin"?
echo $LD_LIBRARY_PATH  # 输出是否包含"/usr/local/cuda/lib64"?
which nvcc             # 输出是否包含"/usr/local/cuda/bin/nvcc"?
nvidia-smi             # 是否显示nvidia的GPU的信息?

安装 cuDNN

https://developer.nvidia.com/rdp/cudnn-download 下载cudnn , 需要登录nvidia账户.才能下载,下载版本需要和上面cuda版本匹配
cuDNN Runtime Library for Ubuntu16.04 (Deb)
cuDNN Developer Library for Ubuntu16.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)
这三个都要下载

# 安装 Runtime library
sudo dpkg -i libcudnn7_*_amd64.deb
# 安装 developer library
sudo dpkg -i libcudnn7-dev_*_amd64.deb
# 安装 code samples and user guide
sudo dpkg -i libcudnn7-doc_*_amd64.deb

确认 CUDA 和 cuDNN 安装成功

/usr/local/cuda/bin/cuda-install-samples-* ~ # 拷贝样板到~目录
cd ~/NVIDIA_CUDA-*_Samples/
make
cd 2_Graphics/volumeRender
./volumeRender

能看到下图则安装成功
2019-07-16-23-29-28屏幕截图.png

「一键投喂 软糖/蛋糕/布丁/牛奶/冰阔乐!」

pch18

(๑>ڡ<)☆谢谢老板~

使用微信扫描二维码完成支付

版权声明:如无特别说明,本文为作者原创,如需转载须联系作者本人同意,未经作者本人同意不得擅自转载。
添加新评论
暂无评论