cuda和cudnn在linux ubuntu16.04下的安装

系统如果一开始就没有装显卡驱动,那么就不用装了,直接装cuda可以顺带把显卡驱动装了。
如果已经装了,那么确保已安装的显卡版本符合cuda版本要求,可以去官网查询。
我个人建议直接不管原来的显卡驱动,先卸载了,直接安装cuda中的显卡驱动即可。
ps:cuda包比较大,里面已经包含了显卡驱动包
cuda安装:
准备工作:
检查自己机器的gpu信息
lspci | grep -i nvidia确认安装了对的内核版本
sudo apt-get install linux-headers-$(uname -r)关闭 nouveau
vim /etc/modprobe.d/blacklist-nouveau.conf
添加如下内容
blacklist nouveau
options nouveau modeset=0
执行如下命令使其生效
sudo update-initramfs -u重启机器,然后输入下面命令查看是否已经关闭nouveaulsmod | grep nouveau重启完机器后,关闭lightdm
如果本身机器就没有进入ui桌面,那么直接运行安装文件即可,如果不是,那么需要先关闭图形界面,操作如下:
ctrl+alt+f1,进入tty1
sudo service lightdm stop运行cuda_xxxxxxxx.run文件
如果只是用来gpu加速用,opengl和x-config都选择no
第一个是选择是否安装显卡驱动,选择yes
然后其他的都选择yes就行了,当然样例也可以选择no,不安装
完成安装后会提示哪些安装成功,如果失败,看下报错信息再重新安装,一般都是没有问题的。
需要返回到ui桌面,操作如下命令:
sudo service lightdm start配置环境变量
vim ~/.bashrc
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH" 
source ~/.bashrccudnn安装:
sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值