ubuntu安装cuda和cudnn最简洁教程

禁用Ubuntu的nouveau:

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

在文件中输入以下下内容:

blacklist nouveau
options nouveau modeset=0
sudo update-initramfs -u

重启

lsmod | grep nouveau

若无内容输出,则禁用成功。

ctrl+alt+F1进入终端界面,关闭图形化界面

sudo service lightdm stop

安装cuda,按q可跳过说明

sudo sh cuda_7.5.18_linux.run

遇到提示是否安装openGL ,选择no(如果你的电脑跟我一样是双显,且主显是非NVIDIA的GPU在工作需要选择no,否则可以yes),其他都选择yes或者默认即可。(如果您的电脑是双显卡且在这一步选择了yes,那么你极有可能安装完CUDA之后,重启图形化界面后遇到登录界面循环问题:输入密码后又跳回密码输入界面。

重新启动图形化界面

sudo service lightdm start 

Alt + ctrl +F7切回图像化界面。
如果出现重复登陆等异常情况,卸载驱动重新安装

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo /usr/bin/nvidia-uninstall

查看安装情况

ls /dev/nvidia*

若结果显示以下下或类似的信息,则安装成功

/dev/nvidia0      /dev/nvidiactl      /dev/nvidia-uvm

若显示以下

ls: cannot access/dev/nvidia*: No such file or directory

或是这样的,只出现

/dev/nvidia0     /dev/nvidiactl

执行

sudo gedit /etc/rc.local

替换文本

#!/bin/bash

/sbin/modprobe nvidia

if [ "$?" -eq 0 ]; then
# Count the number of NVIDIA controllers found.
NVDEVS=`lspci | grep -i NVIDIA`
N3D=`echo "$NVDEVS" | grep "3D controller" | wc -l`
NVGA=`echo "$NVDEVS" | grep "VGA compatible controller" | wc -l`

N=`expr $N3D + $NVGA - 1`
for i in `seq 0 $N`; do
mknod -m 666 /dev/nvidia$i c 195 $i
done

mknod -m 666 /dev/nvidiactl c 195 255

else
exit 1
fi

/sbin/modprobe nvidia-uvm

if [ "$?" -eq 0 ]; then
# Find out the major device number used by the nvidia-uvm driver
D=`grep nvidia-uvm /proc/devices | awk '{print $1}'`

mknod -m 666 /dev/nvidia-uvm c $D 0
else
exit 1
fi
exit 0

重新查看状态

ls /dev/nvidia*

显示如下则成功

/dev/nvidia0       /dev/nvidiactl       /dev/nvidia-uvm 

设置环境变量

 sudo gedit /etc/profile

在打开的文件末尾,添加以下两行

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

测试是否成功,显示信息则成功

source /etc/profile
cat /proc/driver/nvidia/version
nvcc -V   

重启电脑。

安装cudnn,注意版本

sudo dpkg -i libcudnn7_7.5.0.56-1+cuda10.0_amd64.deb
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值