5 Ubuntu 18.04 LTS+GTX1080Ti+CUDA10.0 深度学习主机环境搭建

简介: 由于之前的主机存在运行docker自动重启的问题,跑模型没问题,排查之后没有发现问题所在,所以最后决定升级技嘉主板的bios并重新安装系统和环境。

硬件环境
显卡:GTX1080Ti
内存:32G
硬盘:256G SSD+1T机械
2 安装Ubuntu 18.04 LTS
2.1 下载官方镜像
下载地址:

Download Ubuntu Desktop | Download | Ubuntu

ubuntu.com
2.2 制作USB启动盘
使用Refus烧录镜像到U盘,Refus下载地址:Refus。

其中,持久分区大小可不选,分区类型选择GPT,目标系统类型选择UEFI(需要在bios里开启UEFI启动的功能),最后完成镜像的烧录。

2.3 安装系统
安装系统的步骤不在这里赘述,其他的博客[1]写的很好了,在这里提几个关键点,我的磁盘分区方案是:

2G的EFI分区
32G的swap分区(和内存大小一样)
剩下的ssd存储空间分给根目录
home挂载在机械硬盘下
之后按照步骤直到完成系统的安装。

3 安装显卡驱动
在这一步花费了大量时间,根据网上搜索的结果,大部分博客都是总结了显卡驱动安装的三种方法:

1.使用标准Ubuntu仓库进行自动化安装
2.使用PPA仓库进行自动化安装
3.使用官方的NVIDIA驱动进行手动安装
在尝试这三种方法之后,使用nvidia-smi显示连接不到显卡,重启机器也是直接进入命令行模式。

所以,换了另外一种方法,具体过程如下:

首先,在系统安装完成之后不要换源,执行下面的命令去检查软件的更新

sudo apt-get update
之后在应用列表中打开Software & Updates,点击Additional Drivers,这个时候,官方推荐的列表就会显示了,其实这个跟执行

ubuntu-drivers devices
是一样的显示效果,但是在Software & Updates中操作会更稳妥一些。

点击 Apply Changes,等待进度条走完重启机器就安装完成了。执行nvidia-smi,显示如图,表示成功安装显卡驱动。

4 安装CUDA10.0
下载地址:https://developer.nvidia.com/cuda-toolkit-archive

选择CUDA 10.0

建议下载runfile,安装命令如下:

sudo sh cuda__linux.run
在安装过程中,

选择是否安装显卡驱动时选择否
是否构建/usr/local/cuda软链接时选择是
其他选项按需选择
安装完成后,需要在~/.bashrc(/home/用户名/.bashrc,.表示隐藏文件,直接用vi打开这个文件就行)文件中添加如下语句

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
或者

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
建议使用第二种,因为在构建了/usr/local/cuda软链接之后,我们在.bashrc文件中使用该链接作为环境变量,之后只需修改该软链接就可以切换CUDA的版本。

5 安装cuDNN
cuDNN的下载需要匹配CUDA的版本,也需要适配接下来需要安装的TensorFlow的版本。

cuDNN的下载地址:

cuDNN

developer.nvidia.com
下载cuDNN需要注册英伟达开发者账号,在链接页面中操作即可。

官方安装说明:cuDNN安装指南

其中分为两种:

1.TAR File

tar -xzvf cudnn-9.0-linux-x64-v7.tgz
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*

  1. Debian File

sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-devel_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
确定是否安装好

cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN
执行完成显示Test passed!即可

6 安装深度学习套件
6.1 安装anaconda
官方下载地址:

Anaconda Python/R Distribution - Free Download

www.anaconda.com
图标
安装命令:

bash Anaconda3-5.3.0-Linux-x86_64.sh
安装过程中询问是否将anaconda的安装位置写入.bashrc选择是。

为conda换源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
6.2 安装TensorFlow.
官方教程:

使用 pip 安装 TensorFlow | TensorFlow

tensorflow.google.cn
图标
首先,下载TensorFlow的离线镜像

新建虚拟环境

conda create -n keras pip python=3.6
激活虚拟环境

source activate keras
安装TensorFlow

pip install --ignore-installed --upgrade packageURL
测试

import tensorflow as tf
hello = tf.constant('first tensorflow')
sess = tf.Session()
print sess.run(hello)
6.3 安装keras
在上节的keras虚拟环境中,输入命令:

pip install keras -U --pre
测试

git clone https://github.com/fchollet/keras.git
cd keras/examples/
python mnist_mlp.py
网速较差的可以去对应的git仓库中下载example运行。

7 总结
基本上,深度学习的环境算是搭建完成了,再重新安装docker进行测试以后,运行镜像没有问题。写这篇博客的目的是为了记录自己的安装过程,以备后期使用。如果有写的不合适的地方,欢迎提意见。

目录
相关文章
|
2月前
|
Kubernetes Ubuntu 应用服务中间件
在Ubuntu22.04 LTS上搭建Kubernetes集群
在Ubuntu22.04.4上安装Kubernetes v1.28.7,步骤超详细
340 1
在Ubuntu22.04 LTS上搭建Kubernetes集群
|
Kubernetes 网络协议 Ubuntu
Kubeadm 快速搭建 k8s v1.19.1 集群(Ubuntu Server 20.04 LTS)
安装准备工作安装环境要求:角色 实验环境 生产环境 操作系统 master cpu/内存:2 Core/2G cpu/内存:2 Core/4G linux 内核 4.4+ node cpu/内存:1 Core/2G cpu/内存:4 Core/16G linux 内核 4.4+ 备注 Node:应根据需要运行的容器数量进行配置; Linux 操作系统基于 x86_64 架构的各种 Linux 发行版...
962 2
Kubeadm 快速搭建 k8s v1.19.1 集群(Ubuntu Server 20.04 LTS)
|
6月前
|
Ubuntu 虚拟化
Ubuntu安装VMtools实现与主机之间复制粘贴
一、安装 VMware Tools 右键点击你创建的系统,然后出现菜单下滑找到安装 VMware Tools(T) 这个点击安装; 右键点击你创建的系统,然后出现菜单下滑找到设置; 然后弹出虚拟机设置-->点击选项-->客户机隔离-->启用复制粘贴; 二、Ubuntu命令 这些命令假设你的Ubuntu系统已连接到互联网并且已配置为使用apt软件包管理器; 如果你的网络环境有限,你可能需要提前配置好网络或更换软件源; Ubuntu上安装和配置VMtools以实现与主机之间的文本复制粘贴功能; 1、打开终端(Terminal)。 2、运行以下命令以卸载旧版本的open-vm-tools: su
357 1
|
3月前
|
监控 Ubuntu
如何在 Ubuntu 22.04 LTS 上安装 Logwatch?
如何在 Ubuntu 22.04 LTS 上安装 Logwatch?
29 0
|
3月前
|
Ubuntu Java 数据库
如何在 Ubuntu 22.04 LTS 上安装和配置 OrientDB?
如何在 Ubuntu 22.04 LTS 上安装和配置 OrientDB?
34 1
如何在 Ubuntu 22.04 LTS 上安装和配置 OrientDB?
|
3月前
|
Ubuntu 编译器 C语言
如何在 Ubuntu 22.04 LTS 上安装 Spack?
【1月更文挑战第8天】
69 0
如何在 Ubuntu 22.04 LTS 上安装 Spack?
|
6月前
|
Ubuntu 数据安全/隐私保护
百度搜索:蓝易云【如何在 Ubuntu 22.04 LTS 上安装分区编辑器 GParted?】
希望这些步骤能够帮助您在Ubuntu 22.04 LTS上安装GParted并顺利使用它进行分区管理。
67 0
|
7月前
|
Ubuntu 虚拟化
ubuntu2022.4/kali2022/主机(各种刻录方式)/vmware/virtualbox/(下)
ubuntu2022.4/kali2022/主机(各种刻录方式)/vmware/virtualbox/
35 0
|
7月前
|
Ubuntu Linux 虚拟化
ubuntu2022.4/kali2022/主机(各种刻录方式)/vmware/virtualbox/(上)
ubuntu2022.4/kali2022/主机(各种刻录方式)/vmware/virtualbox/
54 0
|
10月前
|
存储 Ubuntu 安全
在Ubuntu 22.04 LTS Jammy Linux 系统上安装MySQL
在Ubuntu 22.04 LTS Jammy Linux 系统上安装MySQL
1266 0