Upgrade torch(1.0.* -> 1.1.*) and torchvision(0.2.* -> 0.3.*) (#726)

ucbrise · Jun 9, 2019 · b78bbdb · b78bbdb
1 parent ddea39d
commit b78bbdb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/dockerfiles/ClipperDevDockerfile b/dockerfiles/ClipperDevDockerfile
@@ -23,7 +23,7 @@ RUN pip install -q cloudpickle==0.5.* pyzmq==17.0.* requests==2.20.0 subprocess3
   xgboost==0.7.* jsonschema==2.6.* psutil==5.4.* prometheus_client keras==2.2.*
 
 # Install PyTorch
-RUN pip install -q torch==1.0.* torchvision==0.2.*
+RUN pip install -q torch==1.1.* torchvision==0.3.*
 
 # Install kubectl
 RUN curl -LO https://storage.googleapis.com/kubernetes-release/release/$(curl -s https://storage.googleapis.com/kubernetes-release/release/stable.txt)/bin/linux/amd64/kubectl \

diff --git a/dockerfiles/ClipperPy35DevDockerfile b/dockerfiles/ClipperPy35DevDockerfile
@@ -27,7 +27,7 @@ RUN pip3 install cloudpickle==0.5.* pyzmq==17.0.* requests==2.20.0 scikit-learn=
   xgboost==0.7.* urllib3==1.24.* keras==2.2.* # CI is broken when urllib3's version is 1.25.1. Delete urllib3==1.24.* later once version compatibility is stabilized
 
 # Install PyTorch
-RUN pip3 install torch==1.0.* torchvision==0.2.*
+RUN pip3 install torch==1.1.* torchvision==0.3.*
 
 # Install kubectl
 RUN curl -LO https://storage.googleapis.com/kubernetes-release/release/$(curl -s https://storage.googleapis.com/kubernetes-release/release/stable.txt)/bin/linux/amd64/kubectl \

diff --git a/dockerfiles/PyTorchContainerDockerfile b/dockerfiles/PyTorchContainerDockerfile
@@ -5,7 +5,7 @@ FROM ${REGISTRY}/${RPC_VERSION}-rpc:${CODE_VERSION}
 
 LABEL maintainer="Dan Crankshaw <dscrankshaw@gmail.com>"
 
-RUN pip install -q torch==1.0.* torchvision==0.2.*
+RUN pip install -q torch==1.1.* torchvision==0.3.*
 
 COPY containers/python/pytorch_container.py containers/python/container_entry.sh /container/