onnx/nvidia docker: bump versions, fix cuda 12

This commit is contained in:
Koushik Dutta
2024-10-02 09:00:14 -07:00
parent a946be88d3
commit adf4e797c1
5 changed files with 6 additions and 7 deletions

View File

@@ -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

View File

@@ -1,7 +1,7 @@
{
// docker installation
"scrypted.debugHost": "koushik-winvm",
"scrypted.debugHost": "koushik-ubuntuvm",
"scrypted.serverRoot": "/server",
// lxc

View File

@@ -1,12 +1,12 @@
{
"name": "@scrypted/onnx",
"version": "0.1.112",
"version": "0.1.113",
"lockfileVersion": 2,
"requires": true,
"packages": {
"": {
"name": "@scrypted/onnx",
"version": "0.1.112",
"version": "0.1.113",
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
}

View File

@@ -42,5 +42,5 @@
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
},
"version": "0.1.112"
"version": "0.1.113"
}

View File

@@ -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?