diff options
| author |   zhaoyh6 <zhaoyh6@asiainfo.com> | 2022-10-24 16:03:26 +0800 |
|---|---|---|
| committer | zhaoyh6 <zhaoyh6@asiainfo.com> | 2022-10-24 16:03:35 +0800 |
| commit | 7e196b1c74c3f65e46c65246433d4adda47692f7 (patch) | |
| tree | 5f68157162a6172bc47a19b7ae6391379cee5f07 /standalone/src/main/assembly/dockerfile | |
| parent | f0bbd743a618e571dba4192d9afeca8090f46766 (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/dockerfile | 4 |
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 \ |