深度学习环境配置（Ubuntn 16.04+显卡+cuda-9.0+cudnn 7.4.1+pycharm+python3.6)

最新推荐文章于 2025-01-06 10:47:28 发布

原创最新推荐文章于 2025-01-06 10:47:28 发布 · 410 阅读

2 ·

CC 4.0 BY-SA版权

此文章版权归Xinyu_cheng所有, 若有参考，烦请标注

文章标签：

#深度学习环境 #cuda #cudnn

本文详细介绍了如何在Ubuntu 16.04上搭建深度学习环境，包括显卡驱动、CUDA 9.0、cuDNN 7.4.1的安装，以及Python 3.6和PyCharm的配置步骤，适合需要进行深度学习项目开发的读者。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

引言

因为项目需要，搭了很多次深度学习环境，本来写成了一个文档的现在看来没有网页方便，于是就干脆写在csdn上吧，大家也可以看看。
准备工作：

显卡驱动： https://www.geforce.cn/drivers ，根据显卡型号选择所需版本
cuda-9.0： https://developer.nvidia.com/cuda-90-download-archive
cudnn-7.4.1： https://developer.nvidia.com/rdp/cudnn-archive ，需要登录
pycharm：https://www.jetbrains.com/pycharm/download/#section=linux

1 显卡驱动安装

（1）利用下面命令检查自己显卡型号：

lspci | grep -i nvidia

查到自己显卡型号后，下载相应的驱动。
（2）若有驱动存在，打开终端，先删除旧的驱动：

sudo apt-get purge nvidia*

或者

sudo /usr/bin/nvidia-uninstall

（3）禁用自带的 nouveau nvidia驱动（很重要）
编辑文件：

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

并添加如下内容：

blacklist nouveau
options nouveau modeset=0

再更新一下内核：

sudo update-initramfs -u

修改后需要重启系统。确认下Nouveau是已经被你干掉，使用命令：

lsmod | grep nouveau	#无输出则表示禁用成功
sudo service lightdm stop	#停用图形驱动程序，进入命令行模式
sudo chmod +x Nvidia.***.run
sudo ./Nvidia.**.run 	#全部选是，x-server选择yes
sudo service lightdm restart  	#重启图形界面
nvidia-smi 	#出现显卡驱动则表示安装成功

2 cuda编译

利用下列命令安装依赖：

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

安装：

sudo sh cuda.run

显卡驱动选择no，其他选yes
添加环境变量：

gedit ~/.bashrc

在末尾添加：

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新源：

source ~/.bashrc

验证：

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery

返回GPU信息则表示安装成功。

编译错误：
Missing recommended library: libGL.so
解决方法：

sudo find / -name 'libGL*'  
sudo ln -s /usr/lib/libGL.so.1 /usr/lib/libGL.so
sudo ldconfig

3 cudnn安装

tar -zxvf cudnn-9.0-linux-x64-v7.1.tgz  #解压cudnn-9.0-linux-x64-v7.1.tgz当前目录：
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so.7 libcudnn.so		#删除原有动态文件
sudo ln -s libcudnn.so.7.1.4 libcudnn.so.7  
sudo ln -s libcudnn.so.7 libcudnn.so  
sudo ldconfig
nvcc -V 	#有信息则表示成功

4 python 3.6安装

添加仓库：

sudo add-apt-repository ppa:jonathonf/python-3.6

安装：

sudo apt-get update
sudo apt-get install python3.6

5 pycharm安装

解压文件，进入到pycharm/bin文件目录，命令行打开：

sh pycharm.sh