Skip to content

Instantly share code, notes, and snippets.

@zhanghang1989
Last active May 7, 2020 04:18
Show Gist options
  • Save zhanghang1989/e97124f43577cbe1a897fa69517e52a7 to your computer and use it in GitHub Desktop.
Save zhanghang1989/e97124f43577cbe1a897fa69517e52a7 to your computer and use it in GitHub Desktop.
aws s3 cp s3://zhiz-data/datasets/imagenet/ILSVRC2012_img_train.tar .
aws s3 cp s3://zhiz-data/datasets/imagenet/ILSVRC2012_img_val.tar .
cd ResNeSt/
python scripts/dataset/prepare_imagenet.py --download-dir ~/
cd ~/
sudo mkdir -p /media/ramdisk
sudo mount -t tmpfs -o size=200G tmpfs /media/ramdisk
cp -r /home/ubuntu/data/ILSVRC2012/ /media/ramdisk
mkdir ~/data
mkdir ~/data/ILSVRC2012
cd ~/data/ILSVRC2012
aws s3 sync s3://zhiz-data/datasets/imagenets/ .
cd ~/
sudo mkdir -p /media/ramdisk
sudo mount -t tmpfs -o size=200G tmpfs /media/ramdisk
cp -r /home/ubuntu/data/ILSVRC2012/ /media/ramdisk
pip install mxnet-cu100
HOROVOD_GPU_ALLREDUCE=NCCL pip install -v --no-cache-dir horovod
pip install --no-cache mpi4py
aws s3 cp s3://zhiz-data/datasets/imagenet/ILSVRC2012_img_train.tar .
aws s3 cp s3://zhiz-data/datasets/imagenet/ILSVRC2012_img_val.tar .
conda install -y pytorch torchvision cudatoolkit=10.0 -c pytorch
cd ~/
mkdir -p git
cd git/
git clone https://github.com/zhanghang1989/ResNeSt
cd ResNeSt/ && python setup.py develop
python scripts/dataset/prepare_imagenet.py --download-dir ~/