本人的系统环境:
Ubuntu 16.04
Cuda 9.0 + Cudnn 7.0.5
Python 3.7
Anaconda 3
这里推荐大家使用Anaconda,可以比较方便的创建Python虚拟环境,避免不同的Python库之间产生冲突。
一 anaconda安装
anaconda的安装非常简单,可以选择去官网下载,但是因为anaconda的服务器在国外,所以国内的我们在Ubuntu下用pip或者conda安装软件包都容易龟速最终导致失败,但是嗑盐的我们耽误不起时间,所以我们要找一个最好最快的安装的方式,使用清华大学开源软件镜像站 ,下拉到最底端选择Linux,选择最新版(32/64位)下载。
下载完成后,使用bash运行sh脚本文件
bash Anaconda3-5.3.1-Linux-x86_64.sh
接下来,按照提示进行安装
按住回车键一直到进度为100%,阅读完注册信息后,输入“yes”回车继续;
按回车键确认安装位置 /home/{虚拟机名称}/anaconda3 ;
输入“yes”选择添加环境变量;
输入“no”取消安装 VSCode
完成以上步骤之后,anaconda安装成功,可以在终端里输入Python查看一下,因为anaconda的Python会替代Ubuntu的默认python
二 conda创建、查看、删除虚拟环境
1 anaconda创建虚拟环境,创建名称为mmdetection的Python3.7环境
conda create -n mmdetection python=3.7
2 激活虚拟环境,环境名称和上面一致
source activate mmdetection
3 退出虚拟环境
source deactivate mmdetection
4 删除环境,删除之前必须退出虚拟环境
conda remove -n mmdetection --all
如果建立虚拟环境较多,可以使用下面命令查看:
conda-env list
三 conda安装pytorch
pytorch最近已经更新到了稳定版本的1.1,从Pytorch官网上可以看到有多种安装方式:
比较常见的方法是通过pip和conda安装,当使用conda安装的时候,可能会遇到下载过慢的问题,尤其是文件:pytorch-1.0.1-py3.6_cuda90_cudnn7_1.tar.bz2,下载的时候可能会遇到无尽的等待。这里推荐用清华源替代默认conda源的方法,可以解决下载过慢的问题。
清华conda源地址:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
1 添加清华源
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
2 安装pytorch
根据Python和CUDA选择对应的版本,然后官方给出提示可通过运行:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
但是这里一定要注意,去掉-c pytorch,安装的时候才会默认从清华源下载相应的包,因此这里用命令行:
conda install pytorch torchvision cudatoolkit=9.0
安装完成之后,可以进行检查是否安装成功
reference
conda安装Pytorch下载过慢解决办法(3月5日更新ubuntu下pytorch1.0.1安装方法)