diff --git a/.github/workflows/run-docker-image.yml b/.github/workflows/run-docker-image.yml index 5ca70282c..17361cac9 100644 --- a/.github/workflows/run-docker-image.yml +++ b/.github/workflows/run-docker-image.yml @@ -55,8 +55,10 @@ jobs: find / -name libcuda.so* 2>/dev/null # for torch1.13.0-cuda11.6 - cd /opt/conda/lib/stubs && ln -s libcuda.so libcuda.so.1 && cd - - export LD_LIBRARY_PATH=/opt/conda/lib/stubs:$LD_LIBRARY_PATH + if [ -e /opt/conda/lib/stubs/libcuda.so ]; then + cd /opt/conda/lib/stubs && ln -s libcuda.so libcuda.so.1 && cd - + export LD_LIBRARY_PATH=/opt/conda/lib/stubs:$LD_LIBRARY_PATH + fi find / -name libcuda.so* 2>/dev/null echo "LD_LIBRARY_PATH: $LD_LIBRARY_PATH" diff --git a/docker/torch1.9.0-cuda10.2.dockerfile b/docker/torch1.9.0-cuda10.2.dockerfile index 22f949151..faf8b67c6 100644 --- a/docker/torch1.9.0-cuda10.2.dockerfile +++ b/docker/torch1.9.0-cuda10.2.dockerfile @@ -44,7 +44,9 @@ RUN apt-get update && \ zlib1g-dev \ && rm -rf /var/lib/apt/lists/* -RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb | dpkg -i - && \ +RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb && \ + dpkg -i cuda-keyring_1.0-1_all.deb && \ + rm -v cuda-keyring_1.0-1_all.deb && \ apt-get update && \ rm -rf /var/lib/apt/lists/*