mirror of
https://github.com/koush/scrypted.git
synced 2026-02-03 14:13:28 +00:00
onnx/nvidia docker: bump versions, fix cuda 12
This commit is contained in:
@@ -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
|
||||
|
||||
2
plugins/onnx/.vscode/settings.json
vendored
2
plugins/onnx/.vscode/settings.json
vendored
@@ -1,7 +1,7 @@
|
||||
|
||||
{
|
||||
// docker installation
|
||||
"scrypted.debugHost": "koushik-winvm",
|
||||
"scrypted.debugHost": "koushik-ubuntuvm",
|
||||
"scrypted.serverRoot": "/server",
|
||||
|
||||
// lxc
|
||||
|
||||
4
plugins/onnx/package-lock.json
generated
4
plugins/onnx/package-lock.json
generated
@@ -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"
|
||||
}
|
||||
|
||||
@@ -42,5 +42,5 @@
|
||||
"devDependencies": {
|
||||
"@scrypted/sdk": "file:../../sdk"
|
||||
},
|
||||
"version": "0.1.112"
|
||||
"version": "0.1.113"
|
||||
}
|
||||
|
||||
@@ -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?
|
||||
|
||||
Reference in New Issue
Block a user