上一篇我们介绍了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地址,此时弹出我们建立的共享文件夹,如下图:

输入用户名和密码后即可访问。
2、显卡驱动下载

根据自己显卡型号进行选择,然后下载。下载完成后拷贝至我们刚才建立的共享文件夹路径下。
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
弹出如下信息,表示安装成功

二、CUDA安装
(1)cuda下载

根据实际情况选择自己所需的cuda版本进行下载,并拷贝至建立的共享文件夹中。
(2)cuda安装
进入共享文件夹路径,执行安装命令
sudo sh cuda_10.2.89_440.33.01_linux.run (注:实际版本与你下载相一致)
弹出如下界面

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

将第一项清除(按空格,其余默认,然后选择Install)
安装完毕后,相关文件会出现在 /usr/local路径下
(3)cudnn安装
下载地址(需要nvidia账号及密码),选择与自己cuda匹配的版本,如下图

同样将下载完毕的文件拷贝至共享文件夹中,解压
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*
相关文章: