上一篇我们介绍了ubuntuserver18.04的安装,接下来我们介绍在系统安装典型的深度学习框架并进行训练

一,nvidia显卡驱动的安装

1、samba服务器的搭建

由于我们按照的操作系统是server版本,没有图形界面,操作起来相对麻烦,所以我们通过搭建samba服务器,可以在Windows下与ubuntu进行文件的传输。

(a)安装samba服务器

命令:sudo apt-get install samba

(b)配置samba服务器

命令:sudo vim /etc/samba/smb.conf

输入命令后弹出samba服务器配置信息,按“i”键,进入编辑模式,在文档最后添加如下信息

[algserver] ------------------------>对应Windows下共享文件夹名称

path = /home/algserver ------------------------->对应Linux下共享文件夹路径

directory mask = 0777 -------------------------->新创建文件夹属性

valid users = algserver -------------------------->允许访问该共享文件夹的用户

writable = yes --------------------------->读写权限

create mask = 755 ---------------------------->新创建文件属性

注意:确保上述path路径下有对应的共享路径

创建samba账户并设置密码

sudo touch /etc/samba/smbpasswd

sudo smbpasswd -a name (注:name必须为系统中存在的账户)

此时输入密码并确认再次输入一遍

重启samba服务器

命令:sudo service smbd restart

在Windows下输入服务器的ip地址,此时弹出我们建立的共享文件夹,如下图:

在ubuntuserver18.04下安装nvidia驱动及CUDA、cudnn

输入用户名和密码后即可访问。

2、显卡驱动下载

下载地址

在ubuntuserver18.04下安装nvidia驱动及CUDA、cudnn

根据自己显卡型号进行选择,然后下载。下载完成后拷贝至我们刚才建立的共享文件夹路径下。

3、显卡驱动安装

(1)Nouveau禁用(不禁用会影响驱动安装)

输入命令:sudo vim /etc/modprobe.d/blacklist.conf

在末尾添加:blacklist nouveau

更新并重启:sudo update-initramfs -u

sudo reboot

(2)安装驱动

进入放置驱动文件的路径,执行命令

sudo sh NVIDIA-Linux-x86_64-440.64.run

安装过程中忽略CC版本,安装完毕执行如下命令测试显卡

nvidia-smi

弹出如下信息,表示安装成功

在ubuntuserver18.04下安装nvidia驱动及CUDA、cudnn

二、CUDA安装

(1)cuda下载

下载地址

在ubuntuserver18.04下安装nvidia驱动及CUDA、cudnn

根据实际情况选择自己所需的cuda版本进行下载,并拷贝至建立的共享文件夹中。

(2)cuda安装

进入共享文件夹路径,执行安装命令

sudo sh cuda_10.2.89_440.33.01_linux.run (注:实际版本与你下载相一致)

弹出如下界面

在ubuntuserver18.04下安装nvidia驱动及CUDA、cudnn

输入accept (接受相关条款),然后弹出如下信息

在ubuntuserver18.04下安装nvidia驱动及CUDA、cudnn

将第一项清除(按空格,其余默认,然后选择Install)

安装完毕后,相关文件会出现在 /usr/local路径下

(3)cudnn安装

下载地址(需要nvidia账号及密码),选择与自己cuda匹配的版本,如下图

在ubuntuserver18.04下安装nvidia驱动及CUDA、cudnn

同样将下载完毕的文件拷贝至共享文件夹中,解压

tar -xzvf cudnn-10.1-linux-x64-v7.6.0.64.solitairetheme8 (根据下载版本,修改后面的文件名)

将相关文件拷贝至指定目录并修改读写权限

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

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

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

相关文章: