ubuntu 16.04 安装NVIDIA,cuDNN

xiaoxiao2021-02-28  68

cuda安装

官网下载cuda:https://developer.nvidia.com/cuda-downloads,我下载的为cuda-8.0版本

cuda安装步骤如下:

sudo apt-get install mesa-common-dev freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev/etc/modprobe.d/blacklist.conf添加:blacklist amd76x_edacblacklist vga16fbblacklist nouveaublacklist rivafbblacklist nvidiafbblacklist rivatv更新内核: sudo update-initramfs -u卸载旧驱动:sudo apt-get remove --purge nvidia*ctrl+alt+f1进入终端界面。 sudo service lightdm stopsudo sh cuda8.0.27linux.run --no-opengl-libs --verbose 进行安装一路enter和y即可。配置环境变量 在/etc/profile添加:PATH=/usr/local/cuda-8.0/bin:$PATHexport PATH执行 source /etc/profile使其生效添加库搜索路径。在 /etc/ld.so.conf.d/添加文件cuda.conf,内容为:/usr/local/cuda-8.0/lib64执行 sudo ldconfig使其生效执行nvidia-smi测试是否安装成功

cudnn安装:

官网下载cudnn:https://developer.nvidia.com/cudnn

安装cudnn步骤如下:

sudo cp lib* /usr/local/cuda/lib64/

sudo cp cudnn.h /usr/local/cuda/include/

转载请注明原文地址: https://www.6miu.com/read-27046.html

最新回复(0)