Commit 30d7f0b 1 parent c99adce commit 30d7f0b Copy full SHA for 30d7f0b
File tree 2 files changed +2
-2
lines changed
2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -29,7 +29,7 @@ RUN bash ./install_conda.sh && rm install_conda.sh common_utils.sh /opt/conda/re
29
29
30
30
# Install cuda and cudnn
31
31
ARG CUDA_VERSION
32
- RUN wget -q https://raw.githubusercontent.com/pytorch/builder/main/ common/install_cuda.sh -O install_cuda.sh
32
+ COPY ./ common/install_cuda.sh install_cuda.sh
33
33
RUN bash ./install_cuda.sh ${CUDA_VERSION} && rm install_cuda.sh
34
34
ENV DESIRED_CUDA ${CUDA_VERSION}
35
35
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:$PATH
Original file line number Diff line number Diff line change @@ -50,7 +50,7 @@ RUN bash ./install_lcov.sh && rm install_lcov.sh
50
50
51
51
# Install cuda and cudnn
52
52
ARG CUDA_VERSION
53
- RUN wget -q https://raw.githubusercontent.com/pytorch/builder/main/ common/install_cuda.sh -O install_cuda.sh
53
+ COPY ./ common/install_cuda.sh install_cuda.sh
54
54
RUN bash ./install_cuda.sh ${CUDA_VERSION} && rm install_cuda.sh
55
55
ENV DESIRED_CUDA ${CUDA_VERSION}
56
56
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:$PATH
You can’t perform that action at this time.
0 commit comments