ubuntu16.04安装显卡驱动、CUDA、cuDNN

先看看本机的显卡:
spci |grep -i nvidia


一、安装显卡驱动:
1、准备工作
先卸载你电脑中此刻有的nvidia的驱动:
sudo apt-get remove –purge nvidia*

再添加一个PPA到系统,等一下安装驱动要用的。
sudo add-apt-repository ppa:graphics-drivers/ppa

最后sudo apt-get update,重启电脑。


2、安装驱动
CTRL+ATL+F1进入终端1(输入用户名和密码后和终端很相似,可以大胆进去)
sudo service lightdm stop(关闭图形界面) 
sudo apt-get install nvidia-384(自动下载驱动安装,nvidia-384要和显卡型号匹配)
sudo service lightdm start来启动图形界面,并且登录。这个时候应该又回到图形界面。
测试有没安装成功:
nvidia-smi来看是不是能够输出你的GPU的一些信息。要是不能够输出的话,重启。能够输出的话,也建议重启一次。

二、安装CUDA
1、官网下载cuda的驱动,
老版本的下载链接:https://developer.nvidia.com/cuda-toolkit-archive

下载installer type:runfile(local)


2、sudo chmod 777 cuda_8.0.44_linux.run
   sudo ./cuda_8.0.44_linux.run
注意:@有个让你选择是否安装nvidia驱动时,一定要选择否: 
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48? 
因为前面我们已经安装了更加新的nvidia367,所以这里不要选择安装。其余的都直接默认或者选择是即可。

安装完成后,显卡driver版本是9,cuda版本是8。


3、环境变量配置 
打开~/.bashrc文件: sudo gedit ~/.bashrc 
将以下内容写入到~/.bashrc尾部:
export  CUDA_HOME=/usr/local/cuda-8.0
export  PATH=$CUDA_HOME/bin$PATH
export  LD_LIBRARY_PATH=$CUDA_HOME/lib64$LD_LIBRARY_PATH

4、测试CUDA
cuda版本查看:
nvcc --version
samples测试:
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery 
make 
sudo ./deviceQuery 
如果显示一些关于GPU的信息,则说明安装成功。 


三、安装cuDNN
官网下载cuDNN,选择cuDNN vX.X Library for linux下载
下载后执行:sudo tar -zxvf ./cudnn-8.0-linux-x64-v5.1.tgz
 cd cuda; sudo cp lib64/lib* /usr/local/cuda/lib64/; 
   sudo cp include/cudnn.h /usr/local/cuda/include/  
   更新软连接: cd /usr/local/cuda/lib64/
   sudo chmod +r libcudnn.so.5.1.10
   sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5
   sudo ln -sf libcudnn.so.5 libcudnn.so
   sudo ldconfig


注意caffe只支持cuDNN5.0版本
                  tensorflow支持cuDNN6.0版本
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值