美文网首页
深度学习环境配置(Linux)

深度学习环境配置(Linux)

作者: oneLeaf_11 | 来源:发表于2019-02-09 23:43 被阅读0次

    此教程适用于Linux的机器(包括ubuntu16.04,centOS等)

    一:安装cuda

    1)  下载cuda_tool_kit,进入网站后如图 

    图1 cuda 官网

        根据自己的系统版本选择对应的runfile,以ubuntu 16.04为例,下载后应该是一个2GB的安装文件,比如下载到【Downloads】。


    2) 安装

    首先比较重要的一步是需要禁用nouveau驱动,不然安装N卡驱动的时候会失败。步骤如下

    禁用nouveau驱动:

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

    在最后加入下面两行并保存:

    blacklist nouveau

    options nouveau modeset=0

    然后执行:sudo update-initramfs -u

    重启,执行:lsmod | grep nouveau,如果没有输出,表明禁用成功。


    之后禁用x-window服务,安装显卡驱动需要禁用该服务,否则会安装终止,并出现相应需要禁用该服务的错误。 

    sudo service lightdm stop 

    Ctrl-Alt-F1,之后输入用户名密码进入控制台模式,就可以进行安装了


    进入到我们下载的安装文件的文件夹

    cd ~/Downloads

    假设下载的文件的名称是cuda.run

    然后执行安装,命令是:bash cuda.run --no-opengl-libs

    根据提示安装即可,重启电脑,进入终端并输入:nvidia-smi验证是否安装成功。


    二:安装cudnn

    1) 下载cudnn文件,首先进入cuDNN下载(注:需要登录Nvidia账号,注册一下即可),登录账号后如图所示。点击I Agree。。即可看到下载,根据Cuda版本下载,以cuda10为例。

    点击Download cuDNN

    图 cuDNN网站

    进入下载页面

    图 cuDNN 下载

    点击第一个后会出现一个列表,如图所示

    图 cuDNN库文件

    以ubuntu16.04为例,点击“cuDNN Library for Linux” 也就是上图中第三个。

    下载后解压,比如下载到【Dowloads】文件夹,打开终端进入Dowloads,

    假设下载的文件名是cuDNN.tgz,输入命令解压文件

    tar -zxvf cuDNN.tgz

    进入加压出来的cuda文件夹

    cd cuda

    复制文件

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

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

    进入/usr/local/cuda/lib64/

    cd /usr/local/cuda/lib64/

    删掉旧链接,创建新链接

    sudo rm -rf libcudnn.so libcudnn.so.5

    sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5

    sudo ln -s libcudnn.so.5 libcudnn.so

    编辑proflie,添加环境变量

    sudo vim /etc/profile

    进入profile后在最后一行添加以下信息并保存:

    PATH=/usr/local/cuda/bin:$PATH

    export PATH

    创建cuda.conf文件

    sudo vim /etc/ld.so.conf.d/cuda.conf

    在cuda.conf文件中添加以下信息并保存:

    /usr/local/cuda/lib64

    重新加载配置

    sudo ldconfig


    三:安装Anaconda

    1) 下载安装文件,进入Anaconda官网,如图所示

    图 Anaconda 官网然后再在命令行中输入source ~/.bashrc

    根据自己的系统下载安装文件,以Ubuntu16.04为例,点击Linux,根据系统位数下载32位或者64位,应该是以sh结尾的一个安装文件。比如下载到【Dowloads】文件夹


    2) 打开终端,进入Dowloads文件夹,比如你下载的文件名是“Anaconda.sh”,

    输入 “sudo bash Anaconda.sh” 进入安装界面,然后根据提示回车继续,一般安装路径按照默认即可,最后会提示你是否安装一个Micorsoft VScode,选择No即可。

    安装结束后,检查一下环境变量是否被添加,步骤如下

    打开终端,输入“sudo vim ~/.bashrc”

    检查末尾是否已经添加如下变量,没有则添加

    exportPATH="/home/你的用户名/anaconda3/bin:$PATH"

    然后再在命令行中输入source ~/.bashrc

    重新打开一个终端,输入python检查安装是否成功。

    相关文章

      网友评论

          本文标题:深度学习环境配置(Linux)

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