onnx/nvidia docker: bump versions, fix cuda 12

koush · Oct 2, 2024 · adf4e79 · adf4e79
1 parent a946be8
commit adf4e79
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 7 deletions.
diff --git a/install/docker/install-nvidia-graphics.sh b/install/docker/install-nvidia-graphics.sh
@@ -7,7 +7,7 @@ then
         && dpkg -i /cuda-keyring.deb \
         && apt update -q \
         && apt install -y cuda-nvcc-11-8 libcublas-11-8 libcudnn8 cuda-libraries-11-8 \
-        && apt install -y cuda-nvcc-12-4 libcublas-12-4 libcudnn8 cuda-libraries-12-4;
+        && apt install -y cuda-nvcc-12-6 libcublas-12-6 libcudnn9-cuda-12 cuda-libraries-12-6;
 
     if [ "$" != "0" ]
     then

diff --git a/plugins/onnx/.vscode/settings.json b/plugins/onnx/.vscode/settings.json
@@ -1,7 +1,7 @@
 
 {
     // docker installation
-    "scrypted.debugHost": "koushik-winvm",
+    "scrypted.debugHost": "koushik-ubuntuvm",
     "scrypted.serverRoot": "/server",
 
     // lxc

diff --git a/plugins/onnx/package-lock.json b/plugins/onnx/package-lock.json
diff --git a/plugins/onnx/package.json b/plugins/onnx/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.112"
+   "version": "0.1.113"
 }
diff --git a/plugins/onnx/src/requirements.txt b/plugins/onnx/src/requirements.txt
@@ -4,8 +4,7 @@ numpy==1.26.4
 # uncomment to require cuda 12, but most stuff is still targetting cuda 11.
 # however, stuff targetted for cuda 11 can still run on cuda 12.
 # --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
-onnxruntime-gpu==1.18.1; 'linux' in sys_platform and platform_machine != 'aarch64'
-onnxruntime-gpu==1.19.2; 'win32' in sys_platform and platform_machine != 'aarch64'
+onnxruntime-gpu==1.19.2; 'darwin' not in sys_platform and platform_machine != 'aarch64'
 # cpu and coreml execution provider
 onnxruntime; 'darwin' in sys_platform or platform_machine == 'aarch64'
 # nightly?