Allo docker container to use GPU

Signed-off-by: brian <brian@brutex.de>
2023-08-15 20:58:43 +02:00 · 2023-08-15 20:58:43 +02:00 · 74ad5087c1
parent acae3944ec
commit 74ad5087c1
1 changed files with 4 additions and 0 deletions
--- 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
+
+