win10+ubuntu18.04双系统+cuda10+cudnn7.4+tensorRT5深度学习环境

最新推荐文章于 2024-06-28 16:56:23 发布

Arron_hou

最新推荐文章于 2024-06-28 16:56:23 发布

阅读量2.2k

点赞数 4

分类专栏：机器学习文章标签： Ubuntu18.04双系统 cuda cudnn 2080TI 1080Ti

本文链接：https://blog.csdn.net/Arron_hou/article/details/87836760

版权

机器学习专栏收录该内容

16 篇文章 0 订阅

订阅专栏

win10+ubuntu18.04双系统+cuda10+cudnn7.4+tensorRT5深度学习环境

一，首先下载ubuntu18.04。
二，分区
三，制作启动盘
四，安装系统
五，安装显卡驱动
六，安装cuda
七，安装cudnn
八，安装tensorRT
九，安装miniconda（python环境）
十，安装python包
十一，安装pycharm编辑器
十二，其他软件
- 终端神器 oh-my-zsh
- 虚拟终端 screen

在上一篇 win10+ubuntu16.04双系统+cuda+cudnn+tensorRT深度学习环境记录了去年深度学习环境的搭建,随着ubuntu,cuda,tensorflow,pytorch,mxnet版本的更新,重装了系统,同时更加简单.在这里介绍一下.
硬件环境
cpu:i7-7700k
gpu:1080Ti

cpu:i7-8700k
gpu:2080Ti

亲测都没有问题。

一，首先下载ubuntu18.04。

附各版本下载地址
选择 ubuntu-18.04-desktop-amd64.iso 2018-04-26 18:44 1.8G 的最新版本.

二，分区

使用DiskGenius软件将win的一个区分区，预留下几十G空间。最好将系统放于固态中。例如100G固态，分成两个区，第一个区用于win访问，第二个区删掉。留下空白空间。

要是单系统也就不需要分区了.
为了确保双系统启动正常, 一定要先装windows, 再装ubuntu.

三，制作启动盘

使用UltraISO读取下载好的ISO文件。写入U盘中。

四，安装系统

进入bios选择U盘启动,图形化安装。
网上写的实在是不行，
在安装界面选择其他选项
在这里插入图片描述
上面这个图，如果是双系统，会有一个 efi 格式的分区，是window 使用uefi模式进入系统启动项存放的位置，Ubuntu的启动项也要放在那里。网上说的要建/boot分区，都什么年代了，电脑都是uefi模式启动了，以前的方法跟不上时代了。
点击空闲，点击加号，挂载点选择**/ 。如果你内存少于8个G, 建议还是加内存。swag大小可以设置和内存大小一致。
最下面有一个安装启动引导器的设备**，选择 efi 格式的编号，一般后面会跟一个 window boot manager。

五，安装显卡驱动

选择图形化是最靠谱的

更新apt-get源列表

sudo apt-get update
sudo apt-get upgrade
添加驱动源

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

ubuntu搜索软件更新器
在这里插入图片描述
(忽略掉仓库失败…)
点击设置->附加驱动

亲测410比较快,当然也可以415或者你看到的更新的版本.
重启电脑

六，安装cuda

https://developer.download.nvidia.cn/compute/cuda/repos/
https://developer.download.nvidia.cn/compute/machine-learning/repos/

cuda到目前为止最新版本为10, 如果日后更新的话, 要先确定 pytorch 或者 tensorFlow 所适用的版本.

在线安装：
在这里插入图片描述
下载.deb（network）在线安装包。
到下载路径下执行建议的命令

sudo dpkg -i cuda-repo-ubuntu1804_10.1.105-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

执行第一句时,会提示添加apt-key,把提示的命令输一下,继续运行即可.
到这里，执行nvcc -V 提示没有cuda tool,需要将cuda安装路径加入环境变量。
cuda路径可能有所不同，可以cd过去确认。

执行：

sudo  gedit /etc/profile

在文件最后添加:

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda

使环境变量在当前用户下永久有效
执行

reboot

运行

nvcc -V

输出

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

则安装成功

七，安装cudnn

下载地址
在登录，同意License后，有一个下载界面,选择和cuda适配的版本, 同时也要确定 pytorch 或者 tensorFlow 所适用的版本.

选择
在这里插入图片描述

下载之后为.deb文件
通过执行

sudo dpkg -i  libcudnn7_7.4.2.24-1+cuda10.0_amd64.deb
sudo dpkg -i  libcudnn7-dev_7.4.2.24-1+cuda10.0_amd64.deb
sudo dpkg -i  libcudnn7-doc_7.4.2.24-1+cuda10.0_amd64.deb

安装cudnn完成。

八，安装tensorRT

tensorRT的下载地址
同样需要登录
选择
在这里插入图片描述
执行

sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.0-trt5.0.2.6-ga-20181009_1-1_amd64.deb

完成安装

九，安装miniconda（python环境）

清华源下载miniconda3
通过执行

 ./Miniconda3-latest-Linux-x86_64.sh

完成安装
其中可以修改安装路径
安装完成后需要执行

source ~/.bashrc

才能使anaconda的环境变量生效。（当然重启后即生效）
同时增加conda的清华源, 方法见官网.

https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

增加pip清华源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

目前最新的python是3.7版本，但是tensorflow2.0仍然只支持3.6，所以可以通过以下命令将默认环境改为3.6，免去日后不断activate环境的操作。具体适用python3.xx版本, 还是要参考pytorch和tensorFlow 支持的版本.
不管版本高了或者低了, 都可以视同下面的命令修改版本.

conda install python=3.6 --yes

十，安装python包

安装tensorflow-gpu

使用conda安装conda会帮忙下载可以使用的cuda和cudnn,以及cpu并行加速库mkl。目前这样装的还是tensorflow1，推荐使用第二条命令安装tensorflow2。
因为tensorflow跟新太快，附上官方链接：https://tensorflow.google.cn/install/gpu

pip install tensorflow-gpu==2.0.0-alpha0

安装pytorch

在pytorch官网上可以生成安装命令
在这里插入图片描述

conda install pytorch torchvision cuda100 -c pytorch
pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl
pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl

安装mxnet

在mxnet官网上可以生成安装命令
在这里插入图片描述

pip install    mxnet100-cu100mkl

安装jupyter notebook

使用conda安装可以选择conda的不同环境

conda install nb_conda

开启远程访问notebook参考另一篇博客.

十一，安装pycharm编辑器

官方下载
一般社区办就可以
专业版需要付费，当然网上各种方法破解。
随意选择。
推荐申请学生账号，免费使用一年。到期后可以再次申请。有了正版账号可以同步设置，还是很好用的。
在/bin/下直接执行./pycharm.sh就可以运行了。
但是为了更方便，可以创建快捷方式。参考博客

sudo gedit /usr/share/applications/Pycharm.desktop

添加（具体以安装路径为准）

  [Desktop Entry]
  Type=Application
  Name=Pycharm
  GenericName=Pycharm3
  Comment=Pycharm3:The Python IDE
  Exec=sh /home/arron/pycharm-community-2018.2.2/bin/pycharm.sh
  Icon=/home/arron/pycharm-community-2018.2.2/bin/pycharm.png
  Terminal=pycharm
  Categories=Pycharm

搜索pycharm，固定到任务栏中即可。