From 8fda7f5939c135d09286486086a0c04c602636d2 Mon Sep 17 00:00:00 2001 From: Jin Hanyu <48324756+Hanyu-Jin@users.noreply.github.com> Date: Tue, 13 Sep 2022 09:33:58 +0800 Subject: [PATCH] Change the editions of torch and torchvision. (#5717) * add assembly * Enable FL Sever in SGX * Add start-fl-server.py to Docker Image * rm ppml-assebly.xml * Add start-python-fl-server-sgx.sh to Docker Image * Put start-fl-server.py into assembly * Change the editions of torch and torchvision Co-authored-by: Jiaming --- ppml/trusted-big-data-ml/python/docker-gramine/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppml/trusted-big-data-ml/python/docker-gramine/Dockerfile b/ppml/trusted-big-data-ml/python/docker-gramine/Dockerfile index 4b28d3432ae..cbdffc3d577 100644 --- a/ppml/trusted-big-data-ml/python/docker-gramine/Dockerfile +++ b/ppml/trusted-big-data-ml/python/docker-gramine/Dockerfile @@ -208,7 +208,7 @@ RUN env DEBIAN_FRONTEND=noninteractive apt-get update && \ pip install --no-cache-dir numpy scipy && \ pip install --no-cache-dir pandas && \ pip install --no-cache-dir scikit-learn matplotlib seaborn jupyter wordcloud moviepy requests h5py opencv-python protobuf==3.19.0 tensorflow==1.15 && \ - pip install --no-cache-dir torch==1.8.1 torchvision==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html && \ + pip install --no-cache-dir torch==1.5.0 torchvision==0.6.0 -f https://download.pytorch.org/whl/torch_stable.html && \ ln -s /usr/bin/python3 /usr/bin/python && \ # Fix tornado await process pip uninstall -y -q tornado && \