Ubuntu下安装CUDA

前端之家收集整理的这篇文章主要介绍了Ubuntu下安装CUDA前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

原文出处: http://xugaoxiang.com/post/117

软硬件环境

  • ubuntu 16.04 64bit
  • NVidia GTX 1070
  • CUDA 9.0

安装驱动

从官网下载RUN文件,首先屏蔽开源nouvear驱动,编辑/etc/modprobe.d/blacklist-nouveau.conf文件,加入内容

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv

如果你原来安装过驱动,最好先删除

sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove xserver-xorg-video-nouveau

NVIDIA的驱动无法在GUI的环境下安装,所以得切换到命令行,按组合快捷键Ctrl + Alt + F1

sudo /etc/init.d/lightdm stop
sudo ./NVIDIA-Linux-x86_64-390.25.RUN

安装过程基本上就是选中yes or no,装完之后执行

sudo update-initramfs -u

安装CUDA

同样的,我们也进入命令行模式安装cuda

sudo /etc/init.d/lightdm stop
sudo ./cuda_9.0.176_384.81_linux.run
sudo /etc/init.d/lightdm start

然后配置下,编辑~/.bashrc

最后

source ~/.bashrc

安装cudnn

到官网https://developer.nvidia.com/cudnn去下载,这里需要注册,有点麻烦

tar xvf cudnn-9.1-linux-x64-v7.1.tgz
cd cuda
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cd -a cuda/lib64/libcudnn*  /usr/local/cuda-9.0/lib64
sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*

测试

安装完毕,我们来测试下环境是否都ok

刚好我这里跑了示例,所以你能看到gpu基本上是满负荷工作

参考资料

http://www.nvidia.com/Download/index.aspx?lang=en-us
https://developer.nvidia.com/cuda-toolkit
https://blog.csdn.net/hanging_gardens/article/details/64439344

原文链接:https://www.f2er.com/ubuntu/349754.html

猜你在找的Ubuntu相关文章