Skip to content
Permalink
Browse files

Unpin torch-nightly (#1650)

Signed-off-by: Travis Addair <taddair@uber.com>
  • Loading branch information
tgaddair committed Jan 9, 2020
1 parent b505f14 commit 9da2b0197d05f72b949f7497c6f99d5eb9e63beb
Showing with 2 additions and 2 deletions.
  1. +1 −1 Dockerfile.test.cpu
  2. +1 −1 Dockerfile.test.gpu
@@ -104,7 +104,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 --pre torch==1.4.0.dev20200106 ${TORCHVISION_PACKAGE} -v -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html; \
pip install --pre torch ${TORCHVISION_PACKAGE} -v -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html; \
else \
pip install ${PYTORCH_PACKAGE} ${TORCHVISION_PACKAGE} -f https://download.pytorch.org/whl/torch_stable.html; \
fi
@@ -87,7 +87,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 --pre torch==1.4.0.dev20200106 ${TORCHVISION_PACKAGE} -v -f https://download.pytorch.org/whl/nightly/cu${PYTORCH_CUDA}/torch_nightly.html; \
pip install --pre torch ${TORCHVISION_PACKAGE} -v -f https://download.pytorch.org/whl/nightly/cu${PYTORCH_CUDA}/torch_nightly.html; \
else \
pip install ${PYTORCH_PACKAGE} ${TORCHVISION_PACKAGE} -f https://download.pytorch.org/whl/torch_stable.html; \
fi

0 comments on commit 9da2b01

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