ubuntu16.04配置深度学习GPU环境

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_31456593/article/details/81041430

ubuntu16.04配置深度学习GPU环境


一、安装nvidia显卡驱动

1、上英伟达官网下载和自己机子匹配的显卡驱动
https://www.nvidia.com/zh-cn/geforce/

2、禁用nouveau驱动

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

在末尾加上下面语句

blacklist nouveau
options nouveau modeset=0

保存退出

sudo update-initramfs -u

重启
登录命令行界面:ctrl+alt+F1
输入用户名密码
找到驱动所在路径
关掉图形界面
赋予文件执行权限,并sudo运行

sudo chmod 777 NVIDIA-Linux-x86_64-390.67.run
sudo service lightdm stop
sudo ./NVIDIA-Linux-x86_64-390.67.run

安装成功后重启,并检测

nvidia-smi

二、安装cuda9.0
上官网下载cuda9.0
运行安装

三、安装cudnn7
官网下载cudnn7

tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
cd cuda    
sudo cp lib64/lib* /usr/local/cuda/lib64/    
sudo cp include/cudnn.h /usr/local/cuda/include/ 

猜你喜欢

转载自blog.csdn.net/qq_31456593/article/details/81041430