深度学习环境搭建:ubuntu16.04+nvidia驱动+cuda9.0+cudnn7.0.5

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

ubuntu16.04+nvidia驱动+cuda9.0+cudnn7.0.5

安装nvidia驱动

sudo vim /etc/modprobe.d/blacklist.conf
blacklist nouveau
options nouveau modeset=0

sudo update-initramfs -u

reboot

lsmod | grep nouveau
进入命令行界面 ctrl+alt+f1,开始安装

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

reboot


检查gpu安装情况
nvidia-smi


安装cuda

sudo chmod 777 cuda_9.0.176_384.81_linux.run

sudo ./cuda_9.0.176_384.81_linux.run

sudo gedit /etc/profile

打开文件后在文件末尾添加路径,也就是安装目录,命令如下:

export PATH=/usr/local/cuda-9.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH 
source /etc/profile


sudo gedit /etc/ld.so.conf.d/cuda.conf
在打开的文件中添加如下语句:
/usr/local/cuda/lib64
执行sudo ldconfig


安装 cudnn:

tar -zxvf cudnn-9.0-linux-x64-v3.0.29.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.7.3.0 # 自己查看.so的版本
sudo ln -sf libcudnn.so.7.3.0. libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig


猜你喜欢

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