Setup Environment⚓︎
实验环境要求⚓︎
操作系统⚓︎
Ubuntu 18.04 LTS x86_64
注: 感兴趣的windows用户可以尝试使用 windows 10 x64 with Ubuntu18.04 LTS sub-system(Windows 10 Insider Preview build 18975 (Slow) or later for WSL 2)
, 如有任何问题和建议,可以通过 issues
反馈给我们。
本实验中测试均在 Ubuntu 18.04 中完成。
编程语言⚓︎
python3.7.6
(Anaconda3
环境)
学习框架⚓︎
PyTorch==1.5.0
硬件环境(由低到高排序)⚓︎
- 单机
- 单机,单GPU(with CUDA 10.1)
- 单机,多GPU(with CUDA 10.1)
- 多机,多GPU(with CUDA 10.1)
实验环境搭建⚓︎
安装anaconda3⚓︎
安装教程:https://docs.anaconda.com/anaconda/install/linux/#installation
下载地址:https://www.anaconda.com/distribution/#linux
-
下载命令:(注意:$ 为linux系统下的命令提示符,不属于命令部分)
注: 请按照命令行提示安装anaconda,所有选项输入yes)$ wget https://repo.anaconda.com/archive/Anaconda3-2020.02-Linux-x86_64.sh $ bash Anaconda3-2020.02-Linux-x86_64.sh
-
激活conda环境:
$ source ~/.bashrc
-
测试是否安装成功:
若安装成功,则会显示: conda 4.8.2$ conda -V
安装python3.7.6(若使用Anaconda3的base环境,默认是python3.7.6,则不需额外安装)⚓︎
- 创建新的conda环境:
$ conda create -n py37 python=3.7.6
- 激活python3.7:
$ conda activate py37
安装gcc(如果机器已经安装gcc,请忽略)⚓︎
$ sudo apt-get update
$ sudo apt-get install build-essential
安装pytorch (version 1.5.0)⚓︎
CPU版本:
$ conda install pytorch==1.5.0 torchvision==0.6.0 cpuonly -c pytorch
GPU(CUDA10.1)版本:
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch