Skip to content
Permalink
Browse files

Unpined torchvision and updated nightly torch with new command form (#…

  • Loading branch information...
tgaddair committed Aug 30, 2019
1 parent c5b53ec commit 9803bd7d4324e3e77bbcd0e5217b87d9c9012f67
Showing with 6 additions and 6 deletions.
  1. +1 −1 Dockerfile.test.cpu
  2. +1 −1 Dockerfile.test.gpu
  3. +4 −4 docker-compose.test.yml
@@ -106,7 +106,7 @@ RUN python -c "from keras.datasets import mnist; mnist.load_data()"
# Install PyTorch.
RUN pip install future typing
RUN if [[ ${PYTORCH_PACKAGE} == "torch-nightly" ]]; then \
pip install torch_nightly -v -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html; \
pip install --pre torch -v -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html; \
else \
pip install ${PYTORCH_PACKAGE}; \
fi
@@ -88,7 +88,7 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
RUN pip install future typing
RUN if [[ ${PYTORCH_PACKAGE} == "torch-nightly" ]]; then \
PYTORCH_CUDA=$(echo ${CUDA_DOCKER_VERSION} | awk -F- '{print $1}' | sed 's/\.//'); \
pip install torch_nightly -v -f https://download.pytorch.org/whl/nightly/cu${PYTORCH_CUDA}/torch_nightly.html; \
pip install --pre torch -v -f https://download.pytorch.org/whl/nightly/cu${PYTORCH_CUDA}/torch_nightly.html; \
else \
pip install ${PYTORCH_PACKAGE}; \
fi
@@ -236,7 +236,7 @@ services:
TENSORFLOW_PACKAGE: tf-nightly
KERAS_PACKAGE: git+https://github.com/keras-team/keras.git
PYTORCH_PACKAGE: torch-nightly
TORCHVISION_PACKAGE: git+https://github.com/pytorch/vision.git@93bceaf250fe659f4baa9a1ecf79ece70d14b356
TORCHVISION_PACKAGE: git+https://github.com/pytorch/vision.git
MXNET_PACKAGE: mxnet --pre
PYSPARK_PACKAGE: pyspark==2.4.0
test-cpu-openmpi-py3_6-tfhead-kerashead-torchhead-mxnethead-pyspark2_4_0:
@@ -249,7 +249,7 @@ services:
TENSORFLOW_PACKAGE: tf-nightly
KERAS_PACKAGE: git+https://github.com/keras-team/keras.git
PYTORCH_PACKAGE: torch-nightly
TORCHVISION_PACKAGE: git+https://github.com/pytorch/vision.git@93bceaf250fe659f4baa9a1ecf79ece70d14b356
TORCHVISION_PACKAGE: git+https://github.com/pytorch/vision.git
MXNET_PACKAGE: mxnet --pre
PYSPARK_PACKAGE: pyspark==2.4.0
test-cpu-mpich-py3_6-tf1_14_0-keras2_2_4-torch1_1_0-mxnet1_5_0-pyspark2_4_0:
@@ -428,7 +428,7 @@ services:
TENSORFLOW_PACKAGE: tf-nightly-gpu
KERAS_PACKAGE: git+https://github.com/keras-team/keras.git
PYTORCH_PACKAGE: torch-nightly
TORCHVISION_PACKAGE: git+https://github.com/pytorch/vision.git@93bceaf250fe659f4baa9a1ecf79ece70d14b356
TORCHVISION_PACKAGE: git+https://github.com/pytorch/vision.git
MXNET_PACKAGE: mxnet-cu100==1.6.0b20190817 --pre
PYSPARK_PACKAGE: pyspark==2.4.0
test-gpu-openmpi-py3_6-tfhead-kerashead-torchhead-mxnethead-pyspark2_4_0:
@@ -441,7 +441,7 @@ services:
TENSORFLOW_PACKAGE: tf-nightly-gpu
KERAS_PACKAGE: git+https://github.com/keras-team/keras.git
PYTORCH_PACKAGE: torch-nightly
TORCHVISION_PACKAGE: git+https://github.com/pytorch/vision.git@93bceaf250fe659f4baa9a1ecf79ece70d14b356
TORCHVISION_PACKAGE: git+https://github.com/pytorch/vision.git
MXNET_PACKAGE: mxnet-cu100==1.6.0b20190817 --pre
PYSPARK_PACKAGE: pyspark==2.4.0
test-gpu-mpich-py3_6-tf1_14_0-keras2_2_4-torch1_1_0-mxnet1_5_0-pyspark2_4_0:

0 comments on commit 9803bd7

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