small fixes

This commit is contained in:
Fangjun Kuang 2023-07-28 13:37:40 +08:00
parent d73d8acf24
commit d6f3eb84f4
2 changed files with 7 additions and 3 deletions

View File

@ -55,8 +55,10 @@ jobs:
find / -name libcuda.so* 2>/dev/null find / -name libcuda.so* 2>/dev/null
# for torch1.13.0-cuda11.6 # for torch1.13.0-cuda11.6
cd /opt/conda/lib/stubs && ln -s libcuda.so libcuda.so.1 && cd - if [ -e /opt/conda/lib/stubs/libcuda.so ]; then
export LD_LIBRARY_PATH=/opt/conda/lib/stubs:$LD_LIBRARY_PATH 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 find / -name libcuda.so* 2>/dev/null
echo "LD_LIBRARY_PATH: $LD_LIBRARY_PATH" echo "LD_LIBRARY_PATH: $LD_LIBRARY_PATH"

View File

@ -44,7 +44,9 @@ RUN apt-get update && \
zlib1g-dev \ zlib1g-dev \
&& rm -rf /var/lib/apt/lists/* && 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 && \ apt-get update && \
rm -rf /var/lib/apt/lists/* rm -rf /var/lib/apt/lists/*