added docker image (#126)

42741dd4 · Frank Lee · GitHub · 293fb40c · 42741dd4
Unverified Commit 42741dd4 authored Jan 07, 2022 by Frank Lee Committed by GitHub Jan 07, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 0 deletions

docker/Dockerfile docker/Dockerfile +10 -0

No files found.
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
+FROM nvcr.io/nvidia/pytorch:21.07-py3
+
+# install dependencies
+RUN pip install -U pip setuptools wheel \
+ && pip install pytest tensorboard deepspeed apex
+
+# install colossalai
+RUN git clone https://github.com/hpcaitech/ColossalAI.git \
+ && cd ./ColossalAI \
+ && pip install -v --no-cache-dir --global-option="--cuda_ext" .