美文网首页我爱编程
深度学习主机环境搭建cuda9.0+cuDNN7.1.1

深度学习主机环境搭建cuda9.0+cuDNN7.1.1

作者: 亦珩 | 来源:发表于2018-04-04 09:58 被阅读772次

    1 搭建准备


    进行深度学习主机的环境搭建,首先当然是要了解自己的机器啦,下面我们来看看本次搭建的软硬件配置。

    硬件配置:

    机箱主板:联想P910

    存储:512G(SSD)+4T(HDD)+32G(RAM)

    CPU:2*E5-2630V4

    显卡:TITAN XP*1

    软件配置:

    操作系统:64位windows7专业版

    深度学习开发比较方便是在linux系统啦,目前ubuntu16.04是比较稳定的版本,支持大多数的库,亲测不会出现太多的问题。由于配置了显卡和后期会使用CNN,还需要安装cuda和cuDNN,考虑到各版本的相互的兼容性问题,选取比较稳定的软件版本,如下所示:

    ubuntu 16.04

    CUDA 9.0 

    cuDNN 7.1.1

    (各软件及版本下载链接之后给出)

    2 开始搭建


    2.1 安装linux单系统(ubuntu16.04)


    2.1.1 首先去网易开源镜像下载ubuntu16.04

    1.jpg 2.jpg 3.jpg

    2.1.2 下载软碟通,使用软碟通和ubuntu的镜像文件,制作U盘启动盘

    具体操作如下:

    打开-选择下好的系统iso文件-选择u盘-启动-写入硬盘映像

    “写入方式”选择“USB-HDD+”

    接着单击“格式化”,按默认即可(一般默认文件系统位FAT32)

    注意镜像文件大小,应小于U盘大小

    2.1.3 重启电脑,在黑屏状态下,按住F12,进入U盘启动,进入ubuntu安装引导

    ubuntu安装参见链接

    安装时最好保持机器在断网的状态下,提高安装速度

    2.2 安装显卡驱动+CUDA+cuDNN


    2.2.1 下载cuda9.0及安装

    首先,根据硬件型号及系统环境,下载所需资源:(安装cuda时会安装显卡驱动)

    CUDA下载链接+安装方法

    https://developer.nvidia.com/cuda-downloads

    设置环境变量:

    sudo gedit /etc/profile

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

    $ export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH

    2.2.1下载cuDNN及安装

    cuDNN下载链接

    https://developer.nvidia.com/rdp/cudnn-download

    如果是奇怪的后缀,就cp打开

    1sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz

    2sudo cp cuda/include/cudnn.h /usr/local/cuda/include

    3sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

    4sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

    安装tensorflow-GPU版本

    pip install tensorflow-gpu

    相关教程链接:

    https://www.r-bloggers.com/lang/chinese/2042

    Ubuntu16.04 + Titan XP + cuda8.0 + cudnn5.1 + opencv3.3.0 + caffe

    http://blog.csdn.net/mynotwo/article/details/78661664

    Ubuntu16.04安装NVIDIA Titan Xp显卡驱动

    http://blog.csdn.net/javahaoshuang3394/article/details/76425009

    http://www.52nlp.cn/从零开始搭建深度学习服务器环境配置ubuntu-1080ti-cuda-cudnn

    安装完cuda要进行测试

    2、安装cuda 9.0 deb文件(自带显卡驱动)

    2.1 获取CUDA安装包,在NVidia官网下载local deb文件。

    2.2执行如下命令安装:

    $ sudo dpkg -i cuda-repo-ubuntu1410-7-0-local_7.0-28_amd64.deb

    $ sudo apt-get update

    $ sudo apt-get install cuda

    2.3设置环境变量:sudo gedit /etc/profile

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

    $ export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH

    2.4重启机器:用其中一个样例进行验证测试

    $ cuda-install-samples-7.5.sh ~

    $ cd ~/NVIDIA_CUDA-Samples_7.5/5_Simulations/nbody

    $ make

    $ ./nbody

    或者在sample目录中make测试项目1_utility,再运行./deviceQuery

    一些小技巧:

    1.查询NVDIA显卡编号

    lspci | grep -i nvidia

    2.显卡信息

    nvidia-smi

    3.显存信息

    nvidia-smi -l

    3.查询cuda版本+移除cuda+查询cuda链接目录

    apt search cuda

    apt remove cuda

    ll /usr/local/cuda

    ln 可以用来指向cuda目录

    如果是升级cuda和cuDNN,直接覆盖即可,pip会自动进行管理。

    python相关安装:

    sudo apt-get install --no-install-recommends

    git graphviz python-dev python-flask python-flaskext.wtf 

    python-gevent python-h5py python-numpy python-pil 

    python-pip python-protobuf python-scipy

    相关文章

      网友评论

        本文标题:深度学习主机环境搭建cuda9.0+cuDNN7.1.1

        本文链接:https://www.haomeiwen.com/subject/brzlqftx.html