Add dot installation in dockerfile

sensity-ai · Sep 26, 2023 · 7d276ed · 7d276ed
1 parent 249226c
commit 7d276ed
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -35,6 +35,7 @@ RUN conda init bash \
     && cd dot && conda env create -f envs/environment-gpu.yaml \
     && conda activate dot \
     && pip install --no-cache-dir torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118 \
+    && pip install -e . \
     && pip install gdown \
     && gdown 1Qaf9hE62XSvgmxR43dfiwEPWWS_dXSCE \
     && unzip -o dot_model_checkpoints.zip \