美文网首页
深度学习环境配置(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