Skip to content
Permalink
Browse files

Downgrade torchvision to 0.2.2 for non-HEAD builds (#1096)

* Downgrade torchvision to 0.2.2 for non-HEAD builds

Signed-off-by: Alex Sergeev <alsrgv@users.noreply.github.com>

* 0.2.2 -> 0.2.2.post3

Signed-off-by: Alex Sergeev <alsrgv@users.noreply.github.com>

* torchvision==0.3.0 is not compatible with torch-nightly

Signed-off-by: Alex Sergeev <alsrgv@users.noreply.github.com>
  • Loading branch information...
alsrgv committed May 25, 2019
1 parent 26439a1 commit 693e0df34e93262cd83b97c0bc13a8d1913ba878
Showing with 10 additions and 3 deletions.
  1. +2 −1 Dockerfile
  2. +2 −1 Dockerfile.test.cpu
  3. +2 −1 Dockerfile.test.gpu
  4. +4 −0 docker-compose.test.yml
@@ -3,6 +3,7 @@ FROM nvidia/cuda:9.0-devel-ubuntu16.04
# TensorFlow version is tightly coupled to CUDA and cuDNN so it should be selected carefully
ENV TENSORFLOW_VERSION=1.12.0
ENV PYTORCH_VERSION=1.1.0
ENV TORCHVISION_VERSION=0.2.2.post3
ENV CUDNN_VERSION=7.4.1.5-1+cuda9.0
ENV NCCL_VERSION=2.3.7-1+cuda9.0
ENV MXNET_VERSION=1.4.1
@@ -34,7 +35,7 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
rm get-pip.py

# Install TensorFlow, Keras, PyTorch and MXNet
RUN pip install 'numpy<1.15.0' tensorflow-gpu==${TENSORFLOW_VERSION} keras h5py torch==${PYTORCH_VERSION} torchvision mxnet-cu90==${MXNET_VERSION}
RUN pip install 'numpy<1.15.0' tensorflow-gpu==${TENSORFLOW_VERSION} keras h5py torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} mxnet-cu90==${MXNET_VERSION}

# Install Open MPI
RUN mkdir /tmp/openmpi && \
@@ -9,6 +9,7 @@ ARG PYTHON_VERSION=2.7
ARG TENSORFLOW_PACKAGE=tensorflow==1.12.0
ARG KERAS_PACKAGE=keras==2.2.2
ARG PYTORCH_PACKAGE=torch==1.0.0
ARG TORCHVISION_PACKAGE=torchvision==0.2.2.post3
ARG MXNET_PACKAGE=mxnet==1.4.1
ARG PYSPARK_PACKAGE=pyspark==2.4.0

@@ -71,7 +72,7 @@ RUN if [[ ${PYTORCH_PACKAGE} == "torch-nightly" ]]; then \
else \
pip install ${PYTORCH_PACKAGE}; \
fi
RUN pip install torchvision Pillow --no-deps
RUN pip install ${TORCHVISION_PACKAGE} Pillow --no-deps
RUN pip install future typing

# Install MXNet.
@@ -11,6 +11,7 @@ ARG PYTHON_VERSION=2.7
ARG TENSORFLOW_PACKAGE=tensorflow-gpu==1.12.0
ARG KERAS_PACKAGE=keras==2.2.2
ARG PYTORCH_PACKAGE=torch==1.0.0
ARG TORCHVISION_PACKAGE=torchvision==0.2.2.post3
ARG MXNET_PACKAGE=mxnet-cu90==1.4.1
ARG PYSPARK_PACKAGE=pyspark==2.4.0
ARG HOROVOD_BUILD_FLAGS=HOROVOD_GPU_ALLREDUCE=NCCL
@@ -84,7 +85,7 @@ RUN if [[ ${PYTORCH_PACKAGE} == "torch-nightly" ]]; then \
else \
pip install ${PYTORCH_PACKAGE}; \
fi
RUN pip install torchvision Pillow --no-deps
RUN pip install ${TORCHVISION_PACKAGE} Pillow --no-deps
RUN pip install future typing

# Install MXNet.
@@ -104,6 +104,7 @@ services:
TENSORFLOW_PACKAGE: tf-nightly
KERAS_PACKAGE: git+https://github.com/keras-team/keras.git
PYTORCH_PACKAGE: torch-nightly
TORCHVISION_PACKAGE: torchvision_nightly
PYSPARK_PACKAGE: pyspark==2.4.0
MXNET_PACKAGE: mxnet --pre
test-cpu-openmpi-py3_6-tfhead-kerashead-torchhead-mxnethead-pyspark2_4_0:
@@ -115,6 +116,7 @@ services:
TENSORFLOW_PACKAGE: tf-nightly
KERAS_PACKAGE: git+https://github.com/keras-team/keras.git
PYTORCH_PACKAGE: torch-nightly
TORCHVISION_PACKAGE: torchvision_nightly
PYSPARK_PACKAGE: pyspark==2.4.0
MXNET_PACKAGE: mxnet --pre
test-cpu-mpich-py2_7-tf1_12_0-keras2_2_2-torch1_0_0-mxnet1_4_1-pyspark2_4_0:
@@ -192,6 +194,7 @@ services:
TENSORFLOW_PACKAGE: tf-nightly-gpu==1.14.1.dev20190517
KERAS_PACKAGE: git+https://github.com/keras-team/keras.git
PYTORCH_PACKAGE: torch-nightly
TORCHVISION_PACKAGE: torchvision_nightly
MXNET_PACKAGE: mxnet-cu100 --pre
PYSPARK_PACKAGE: pyspark==2.4.0
test-gpu-openmpi-py3_6-tfhead-kerashead-torchhead-mxnethead-pyspark2_4_0:
@@ -206,6 +209,7 @@ services:
TENSORFLOW_PACKAGE: tf-nightly-gpu==1.14.1.dev20190517
KERAS_PACKAGE: git+https://github.com/keras-team/keras.git
PYTORCH_PACKAGE: torch-nightly
TORCHVISION_PACKAGE: torchvision_nightly
MXNET_PACKAGE: mxnet-cu100 --pre
PYSPARK_PACKAGE: pyspark==2.4.0
test-gpu-mpich-py2_7-tf1_12_0-keras2_2_2-torch1_0_0-mxnet1_4_1-pyspark2_4_0:

0 comments on commit 693e0df

Please sign in to comment.
You can’t perform that action at this time.