Update pytorch base.Dockerfile

js-ish · Jul 8, 2023 · 4242087 · 4242087
1 parent 8f4ad23
commit 4242087
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pytorch/base.Dockerfile b/pytorch/base.Dockerfile
@@ -11,7 +11,8 @@ RUN set -eux; \
     ; \
     rm -rf /var/lib/apt/lists/*
 
-RUN pip install --disable-pip-version-check --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu116 \
-    torch==1.13.1+cu116 torchvision==0.14.1+cu116; \
+RUN pip install --disable-pip-version-check --no-cache-dir \
+       torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 torchtext==0.14.1 \
+       --extra-index-url https://download.pytorch.org/whl/cu116; \
     pip install --disable-pip-version-check --no-cache-dir \
-    torchaudio==0.13.1 torchtext==0.14.1 Pillow scikit-learn
+       Pillow scikit-learn