mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-03-10 02:29:19 +03:00
Update onnx deps to support 50 series GPUs (#22324)
* Update support for newer Nvidia GPUs * Update * Cleanup * Update gitignore
This commit is contained in:
parent
a705f254e5
commit
ef07563d0a
2
.gitignore
vendored
2
.gitignore
vendored
@ -3,6 +3,8 @@ __pycache__
|
|||||||
.mypy_cache
|
.mypy_cache
|
||||||
*.swp
|
*.swp
|
||||||
debug
|
debug
|
||||||
|
.claude/*
|
||||||
|
.mcp.json
|
||||||
.vscode/*
|
.vscode/*
|
||||||
!.vscode/launch.json
|
!.vscode/launch.json
|
||||||
config/*
|
config/*
|
||||||
|
|||||||
@ -1,18 +1,18 @@
|
|||||||
# NVidia TensorRT Support (amd64 only)
|
# Nvidia ONNX Runtime GPU Support
|
||||||
--extra-index-url 'https://pypi.nvidia.com'
|
--extra-index-url 'https://pypi.nvidia.com'
|
||||||
cython==3.0.*; platform_machine == 'x86_64'
|
cython==3.0.*; platform_machine == 'x86_64'
|
||||||
nvidia_cuda_cupti_cu12==12.5.82; platform_machine == 'x86_64'
|
nvidia-cuda-cupti-cu12==12.9.79; platform_machine == 'x86_64'
|
||||||
nvidia-cublas-cu12==12.5.3.*; platform_machine == 'x86_64'
|
nvidia-cublas-cu12==12.9.1.*; platform_machine == 'x86_64'
|
||||||
nvidia-cudnn-cu12==9.3.0.*; platform_machine == 'x86_64'
|
nvidia-cudnn-cu12==9.19.0.*; platform_machine == 'x86_64'
|
||||||
nvidia-cufft-cu12==11.2.3.*; platform_machine == 'x86_64'
|
nvidia-cufft-cu12==11.4.1.*; platform_machine == 'x86_64'
|
||||||
nvidia-curand-cu12==10.3.6.*; platform_machine == 'x86_64'
|
nvidia-curand-cu12==10.3.10.*; platform_machine == 'x86_64'
|
||||||
nvidia_cuda_nvcc_cu12==12.5.82; platform_machine == 'x86_64'
|
nvidia-cuda-nvcc-cu12==12.9.86; platform_machine == 'x86_64'
|
||||||
nvidia-cuda-nvrtc-cu12==12.5.82; platform_machine == 'x86_64'
|
nvidia-cuda-nvrtc-cu12==12.9.86; platform_machine == 'x86_64'
|
||||||
nvidia_cuda_runtime_cu12==12.5.82; platform_machine == 'x86_64'
|
nvidia-cuda-runtime-cu12==12.9.79; platform_machine == 'x86_64'
|
||||||
nvidia_cusolver_cu12==11.6.3.*; platform_machine == 'x86_64'
|
nvidia-cusolver-cu12==11.7.5.*; platform_machine == 'x86_64'
|
||||||
nvidia_cusparse_cu12==12.5.1.*; platform_machine == 'x86_64'
|
nvidia-cusparse-cu12==12.5.10.*; platform_machine == 'x86_64'
|
||||||
nvidia_nccl_cu12==2.23.4; platform_machine == 'x86_64'
|
nvidia-nccl-cu12==2.29.7; platform_machine == 'x86_64'
|
||||||
nvidia_nvjitlink_cu12==12.5.82; platform_machine == 'x86_64'
|
nvidia-nvjitlink-cu12==12.9.86; platform_machine == 'x86_64'
|
||||||
onnx==1.16.*; platform_machine == 'x86_64'
|
onnx==1.16.*; platform_machine == 'x86_64'
|
||||||
onnxruntime-gpu==1.22.*; platform_machine == 'x86_64'
|
onnxruntime-gpu==1.24.*; platform_machine == 'x86_64'
|
||||||
protobuf==3.20.3; platform_machine == 'x86_64'
|
protobuf==3.20.3; platform_machine == 'x86_64'
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user