aboutsummaryrefslogtreecommitdiffstats
path: root/standalone/src/main/assembly/dockerfile
diff options
context:
space:
mode:
authorzhaoyh6 <zhaoyh6@asiainfo.com>2022-10-24 16:03:26 +0800
committerzhaoyh6 <zhaoyh6@asiainfo.com>2022-10-24 16:03:35 +0800
commit7e196b1c74c3f65e46c65246433d4adda47692f7 (patch)
tree5f68157162a6172bc47a19b7ae6391379cee5f07 /standalone/src/main/assembly/dockerfile
parentf0bbd743a618e571dba4192d9afeca8090f46766 (diff)
feat:Integrated TenserFlow/PyTorch dual framework.1.0.4
Issue-ID: REQ-1214 Signed-off-by: zhaoyh6 <zhaoyh6@asiainfo.com> Change-Id: I25e78bbe2c524ea9dca8f39387178fb95fb74ffb
Diffstat (limited to 'standalone/src/main/assembly/dockerfile')
-rw-r--r--standalone/src/main/assembly/dockerfile4
1 files changed, 4 insertions, 0 deletions
diff --git a/standalone/src/main/assembly/dockerfile b/standalone/src/main/assembly/dockerfile
index 185d25b..afa4e9a 100644
--- a/standalone/src/main/assembly/dockerfile
+++ b/standalone/src/main/assembly/dockerfile
@@ -5,6 +5,8 @@ WORKDIR /home/uuihome/uui
COPY requirements.txt /home/uuihome/uui/requirements.txt
+RUN apt-get update && apt-get install -y ruby zlib1g zlib1g.dev libjpeg-dev
+
# hadolint ignore=DL3008
RUN groupadd uui -g 1000 && \
useradd -u 1000 -d /home/uuihome/uui -g uui -s /bin/bash uui && \
@@ -23,6 +25,8 @@ RUN groupadd uui -g 1000 && \
ln -s /usr/bin/python3 /usr/bin/python && \
ln -s /usr/bin/pip3 /usr/bin/pip && \
pip install --upgrade setuptools && \
+ pip install wheel && \
+ pip install torch==1.10.1 torchvision==0.11.2+cpu torchaudio==0.10.1 --extra-index-url https://download.pytorch.org/whl/cpu && \
pip install --no-cache-dir -r requirements.txt && \
DEBIAN_FRONTEND=noninteractive apt-get remove --purge -y \
build-essential \