From be7cd6b9303abc8642b3a959ff3c819ac843bbc5 Mon Sep 17 00:00:00 2001
From: brian <brian@brutex.de>
Date: Tue, 15 Aug 2023 20:25:35 +0200
Subject: [PATCH] Allo docker container to use GPU

Signed-off-by: brian <brian@brutex.de>
---
 .docker/Dockerfile                         | 1 +
 cavis-common-platform/build.gradle         | 1 +
 cavis-native/cavis-native-lib/build.gradle | 2 +-
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.docker/Dockerfile b/.docker/Dockerfile
index 766b86554..dbd2a68c2 100644
--- a/.docker/Dockerfile
+++ b/.docker/Dockerfile
@@ -10,6 +10,7 @@ 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
 
diff --git a/cavis-common-platform/build.gradle b/cavis-common-platform/build.gradle
index 7b712082e..52c6b4b34 100644
--- a/cavis-common-platform/build.gradle
+++ b/cavis-common-platform/build.gradle
@@ -130,6 +130,7 @@ dependencies {
         api "org.bytedeco:cuda:${cuda.version}-${cudnn.version}-${javacpp.presetsVersion}"
         api "org.bytedeco:cuda-platform-redist:${cuda.version}-${cudnn.version}-${javacpp.presetsVersion}"
         api "org.bytedeco:mkl-dnn:0.21.5-${javacpp.presetsVersion}"
+        api "org.bytedeco:mkl:2022.0-${javacpp.presetsVersion}"
         api "org.bytedeco:tensorflow:${tensorflow.version}-${javacpp.presetsVersion}"
         api "org.bytedeco:cpython:${cpython.version}-${javacpp.presetsVersion}:${javacppPlatform}"
         api "org.bytedeco:numpy:${numpy.version}-${javacpp.presetsVersion}:${javacppPlatform}"
diff --git a/cavis-native/cavis-native-lib/build.gradle b/cavis-native/cavis-native-lib/build.gradle
index 4369274b1..70ebf5418 100644
--- a/cavis-native/cavis-native-lib/build.gradle
+++ b/cavis-native/cavis-native-lib/build.gradle
@@ -64,7 +64,7 @@ buildscript {
 
 plugins {
     id 'java-library'
-    id 'org.bytedeco.gradle-javacpp-build' version "1.5.9"
+    id 'org.bytedeco.gradle-javacpp-build' version "1.5.7"
     id 'maven-publish'
     id 'signing'
 }