From 42741dd4a37aa9517737681418fd834f0f885dc3 Mon Sep 17 00:00:00 2001 From: Frank Lee Date: Fri, 7 Jan 2022 14:54:04 +0800 Subject: [PATCH] added docker image (#126) --- docker/Dockerfile | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 docker/Dockerfile diff --git a/docker/Dockerfile b/docker/Dockerfile new file mode 100644 index 000000000..a58e8dc61 --- /dev/null +++ b/docker/Dockerfile @@ -0,0 +1,10 @@ +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" .