From eec8b0ad98a668b88979b3e44716234e05a52410 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Thu, 27 Jun 2024 18:30:28 -0400 Subject: [PATCH] Export envs --- docker/gcp-a100-runner-dind.dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/gcp-a100-runner-dind.dockerfile b/docker/gcp-a100-runner-dind.dockerfile index 5d39eb686..59c3cea99 100644 --- a/docker/gcp-a100-runner-dind.dockerfile +++ b/docker/gcp-a100-runner-dind.dockerfile @@ -33,7 +33,7 @@ RUN sudo mkdir -p /workspace; sudo chown runner:runner /workspace # Use the CUDA installation scripts from pytorch/builder # Install CUDA 12.4 only to reduce docker size RUN cd /workspace; git clone https://github.com/pytorch/builder.git -RUN sudo bash -c "set -x; source /workspace/builder/common/install_cuda.sh; install_124; prune_124" +RUN sudo bash -c "set -x; source /workspace/builder/common/install_cuda.sh; install_124; export OVERRIDE_GENCODE=\"${OVERRIDE_GENCODE}\" OVERRIDE_GENCODE_CUDNN=\"${OVERRIDE_GENCODE_CUDNN}\"; prune_124" # Install miniconda RUN wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /workspace/Miniconda3-latest-Linux-x86_64.sh