Ubuntu 18.04 下深度学习环境配置

Ubuntu 18.04 下深度学习环境配置

首先可以参考我的上一篇博客安装好Ubuntu 18.04：Ubuntu 18.04常用软件安装与设置
下文参考链接：NVIDIA显卡深度学习环境配置教程

1. 安装显卡驱动

sudo add-apt-repository ppa:xorg-edgers/ppa #添加ppa源
sudo add-apt-repository ppa:graphics-drivers/ppa #添加ppa源
sudo apt-get update #更新apt-get
sudo apt-get install nvidia-drivers-410 #安装410版本的driver

用以下命令进行测试：

nvidia-smi #若列出GPU的信息列表，表示驱动安装成功
nvidia-settings #若弹出设置对话框，亦表示驱动安装成功

2. 安装cuda与cudnn

详见参考链接，注意各个安装包版本对应

3. 安装MiniConda与pytorch环境

• Miniconda安装
  参考链接：Anaconda介绍、安装及使用教程
  下载链接：https://docs.conda.io/en/latest/miniconda.html
  下载对应版本并运行bash Miniconda*.sh进行安装
  在.bashrc文件中加入conda deactivate可以去掉终端中的(base)
• 创建虚拟环境并安装pytorch
  conda create -n pytorch python=3.7 numpy
  进入pytorch官网选择对应版本，运行对应命令，并指定清华的源

conda install pytorch torchvision cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

4. 安装OpenCV

参考链接：Ubuntu 16.04 安装opencv的各种方法

• 编译cpp版本可能遇到的问题

1. 与cuda有关：……

• 安装完后验证是否成功，查看opencv版本：pkg-config --modversion opencv

  5. 安装darknet

  参考链接：YOLOv3