此教程适用于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检查安装是否成功。
网友评论