update

pytorch · Mar 8, 2024 · 905dc6e · 905dc6e
1 parent dba2b77
commit 905dc6e
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/.ci/docker/Dockerfile b/.ci/docker/Dockerfile
@@ -31,17 +31,10 @@ RUN bash ./install_conda.sh && rm install_conda.sh common_utils.sh /opt/conda/re
 # COPY ./install_gcc.sh install_gcc.sh
 # RUN bash ./install_gcc.sh && rm install_gcc.sh
 
-# Install cuda
-RUN wget -q https://raw.githubusercontent.com/pytorch/builder/main/common/install_cuda.sh -O install_cuda.sh
-RUN bash ./install_cuda.sh ${CUDA_VERSION} && rm install_cuda.sh
-ENV DESIRED_CUDA ${CUDA_VERSION}
-ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:$PATH
-
 # Install CUDNN
 ENV CUDNN_VERSION=8
 COPY ./common/install_cudnn.sh install_cudnn.sh
-RUN if [ "${CUDNN_VERSION}" -eq 8 ]; then bash install_cudnn.sh; fi
-RUN rm install_cudnn.sh
+RUN bash ./install_cudnn.sh && rm install_cudnn.sh
 
 USER jenkins
 CMD ["bash"]