From 4c01abbab132cc1c9e15fa7e4372e9af845fb0ef Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Fri, 9 Aug 2024 15:46:21 -0400 Subject: [PATCH] fix dockerfile --- docker/Dockerfile | 6 ------ 1 file changed, 6 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 9e86c08fd3..970af2f1ef 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -189,15 +189,9 @@ ENV PYTORCH_NIGHTLY_VERSION=${PYTORCH_NIGHTLY_VERSION} RUN if [ -z "$PYTORCH_NIGHTLY_URL" ] ; then \ CUDA_VERSION_TAG=$(python${PYTHON_VERSION} -c "print('cu' + ''.join('${CUDA_VERSION}'.split('.')[:2]) if '${CUDA_VERSION}' else 'cpu')") && \ - if [ "${PYTORCH_VERSION}" != "2.4.0" ]; then \ pip${PYTHON_VERSION} install --no-cache-dir --find-links https://download.pytorch.org/whl/torch_stable.html \ torch==${PYTORCH_VERSION}+${CUDA_VERSION_TAG} \ torchvision==${TORCHVISION_VERSION}+${CUDA_VERSION_TAG} ; \ - else \ - pip${PYTHON_VERSION} install --no-cache-dir --find-links https://download.pytorch.org/whl/${CUDA_VERSION} \ - torch \ - torchvision ; \ - fi ; \ else \ pip${PYTHON_VERSION} install --no-cache-dir --pre --index-url ${PYTORCH_NIGHTLY_URL} \ torch==${PYTORCH_VERSION}.${PYTORCH_NIGHTLY_VERSION} \