From 74ad5087c1ec98e61fb6f1d6e92819cdce5adfe8 Mon Sep 17 00:00:00 2001 From: brian Date: Tue, 15 Aug 2023 20:58:43 +0200 Subject: [PATCH] Allo docker container to use GPU Signed-off-by: brian --- .docker/Dockerfile | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.docker/Dockerfile b/.docker/Dockerfile index dbd2a68c2..79b0ae61c 100644 --- a/.docker/Dockerfile +++ b/.docker/Dockerfile @@ -10,7 +10,11 @@ RUN wget -nv https://github.com/Kitware/CMake/releases/download/v3.24.2/cmake-3. mkdir /opt/cmake && sh ./cmake-3.24.2-linux-x86_64.sh --skip-license --prefix=/opt/cmake && ln -s /opt/cmake/bin/cmake /usr/bin/cmake && \ rm cmake-3.24.2-linux-x86_64.sh + RUN echo "/usr/local/cuda/compat/" >> /etc/ld.so.conf.d/cuda-driver.conf RUN echo "nameserver 8.8.8.8" >> /etc/resolv.conf +RUN ldconfig -p | grep cuda + +