Ubuntu 16.04深度学习环境搭建全指南

📅 发布时间:2026/7/19 7:16:27
Ubuntu 16.04深度学习环境搭建全指南 1. 环境准备与依赖项安装在Ubuntu 16.04系统上搭建深度学习开发环境需要先安装必要的依赖项。这些基础组件将为后续的软件安装提供支持。首先更新系统软件包列表sudo apt-get update sudo apt-get upgrade -y安装通用依赖项sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev \ libopencv-dev libhdf5-serial-dev protobuf-compiler \ libboost-all-dev libgflags-dev libgoogle-glog-dev \ liblmdb-dev libatlas-base-dev注意安装过程中如果遇到依赖冲突可以尝试使用--fix-missing参数自动修复依赖关系。2. CUDA 8.0安装与配置2.1 驱动准备在安装CUDA之前需要确保系统已安装合适的NVIDIA驱动sudo apt-get install nvidia-384安装完成后重启系统通过nvidia-smi命令验证驱动是否正常工作。2.2 CUDA安装从NVIDIA官网下载CUDA 8.0的runfile安装包wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run执行安装sudo sh cuda_8.0.61_375.26_linux-run安装过程中需要注意不安装自带的NVIDIA驱动已单独安装接受许可协议保持默认安装路径(/usr/local/cuda-8.0)2.3 环境变量配置在~/.bashrc文件中添加export PATH/usr/local/cuda-8.0/bin:$PATH export LD_LIBRARY_PATH/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH执行source ~/.bashrc使配置生效通过nvcc -V验证安装。3. cuDNN 6.0安装从NVIDIA开发者网站下载cuDNN 6.0 for CUDA 8.0的安装包解压后执行sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod ar /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*4. OpenCV 3.1编译安装4.1 源码准备从OpenCV官网下载3.1.0版本源码并解压wget https://github.com/opencv/opencv/archive/3.1.0.zip unzip 3.1.0.zip cd opencv-3.1.04.2 编译配置创建build目录并配置mkdir build cd build cmake -D CMAKE_BUILD_TYPERELEASE \ -D CMAKE_INSTALL_PREFIX/usr/local \ -D WITH_CUDAON \ -D CUDA_ARCH_BIN3.0 3.5 5.0 6.0 \ -D CUDA_ARCH_PTX \ -D WITH_CUBLASON \ -D WITH_LIBV4LON \ -D BUILD_TESTSOFF \ -D BUILD_PERF_TESTSOFF ..4.3 解决GraphCut问题编辑modules/cudalegacy/src/graphcuts.cpp文件修改#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER) 为 #if !defined (HAVE_CUDA) || defined (CUDA_DISABLER) || (CUDART_VERSION 8000)4.4 编译与安装make -j$(nproc) sudo make install sudo ldconfig5. MATLAB 2017a安装5.1 挂载ISO镜像sudo mkdir /media/matlab sudo mount -o loop R2017a_glnxa64.iso /media/matlab5.2 执行安装cd /media/matlab sudo ./install安装过程中选择使用文件安装密钥模式输入许可证文件中的安装密钥。5.3 破解与配置复制破解文件到安装目录sudo cp -r Matlab_R2017a_Linux_Crack/R2017a /usr/local/MATLAB/创建桌面快捷方式sudo ln -s /usr/local/MATLAB/R2017a/bin/matlab /usr/local/bin/matlab6. Caffe编译与配置6.1 获取源码git clone https://github.com/BVLC/caffe.git cd caffe cp Makefile.config.example Makefile.config6.2 配置文件修改编辑Makefile.config关键配置如下USE_CUDNN : 1 OPENCV_VERSION : 3 CUDA_DIR : /usr/local/cuda-8.0 BLAS : atlas MATLAB_DIR : /usr/local/MATLAB/R2017a PYTHON_INCLUDE : /usr/include/python2.7 \ /usr/lib/python2.7/dist-packages/numpy/core/include INCLUDE_DIRS : $(PYTHON_INCLUDE) /usr/local/include /usr/include \ /usr/lib/x86_64-linux-gnu/hdf5/serial/include LIBRARY_DIRS : $(PYTHON_LIB) /usr/local/lib /usr/lib \ /usr/lib/x86_64-linux-gnu/hdf5/serial LINKFLAGS : -Wl,-rpath,$(HOME)/anaconda2/lib6.3 编译安装make all -j$(nproc) make test -j$(nproc) make runtest -j$(nproc) make pycaffe -j$(nproc) make matcaffe -j$(nproc)6.4 常见问题解决问题1libcudnn找不到sudo ln -s /usr/local/cuda/lib64/libcudnn.so.6 /usr/local/cuda/lib64/libcudnn.so问题2hdf5相关错误在Makefile.config中确保包含INCLUDE_DIRS : ... /usr/include/hdf5/serial LIBRARY_DIRS : ... /usr/lib/x86_64-linux-gnu/hdf5/serial问题3protobuf缺失conda install protobuf7. 环境验证7.1 Python接口测试import caffe print(caffe.__version__)7.2 MATLAB接口测试在MATLAB命令行执行addpath(/path/to/caffe/matlab); caffe.set_mode_cpu();7.3 MNIST测试./data/mnist/get_mnist.sh ./examples/mnist/create_mnist.sh ./examples/mnist/train_lenet.sh8. 使用技巧与优化建议多版本管理使用virtualenv或conda创建独立Python环境性能调优在Makefile.config中调整CUDA架构参数内存优化对于小显存GPU可减小batch size开发工具推荐使用PyCharm专业版进行Python开发MATLAB优化设置MATLAB使用GPU计算gpuDevice()9. 系统维护建议定期清理编译中间文件make clean更新驱动和CUDA时注意版本兼容性重要配置文件备份tar -czvf caffe_config_backup.tar.gz Makefile.config使用screen或tmux管理长时间运行的任务这套环境配置虽然过程复杂但完成后可以支持大多数深度学习研究和开发需求。在实际使用中建议根据具体项目需求调整各组件版本。