From 424208785985678d1b95b63ee198875828c94f5a Mon Sep 17 00:00:00 2001
From: Michael Ding <yandy.ding@gmail.com>
Date: Sat, 8 Jul 2023 13:57:54 +0800
Subject: [PATCH] Update pytorch base.Dockerfile

---
 pytorch/base.Dockerfile | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/pytorch/base.Dockerfile b/pytorch/base.Dockerfile
index 0d256f0..b378f31 100644
--- 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