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 <[email protected]>
intel-analytics · Sep 13, 2022 · 8fda7f5 · 8fda7f5
1 parent 6509bd6
commit 8fda7f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 && \