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版本可能遇到的问题
- 与cuda有关:……
- 安装完后验证是否成功,查看opencv版本:
pkg-config --modversion opencv
5. 安装darknet
参考链接:YOLOv3