diff --git a/.gitignore b/.gitignore index 660a378b0..c9db2929f 100644 --- a/.gitignore +++ b/.gitignore @@ -3,6 +3,8 @@ __pycache__ .mypy_cache *.swp debug +.claude/* +.mcp.json .vscode/* !.vscode/launch.json config/* @@ -19,4 +21,4 @@ web/.env core !/web/**/*.ts .idea/* -.ipynb_checkpoints \ No newline at end of file +.ipynb_checkpoints diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index 63c68b583..caa65cdb6 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -1,18 +1,18 @@ -# NVidia TensorRT Support (amd64 only) +# Nvidia ONNX Runtime GPU Support --extra-index-url 'https://pypi.nvidia.com' cython==3.0.*; platform_machine == 'x86_64' -nvidia_cuda_cupti_cu12==12.5.82; platform_machine == 'x86_64' -nvidia-cublas-cu12==12.5.3.*; platform_machine == 'x86_64' -nvidia-cudnn-cu12==9.3.0.*; platform_machine == 'x86_64' -nvidia-cufft-cu12==11.2.3.*; platform_machine == 'x86_64' -nvidia-curand-cu12==10.3.6.*; platform_machine == 'x86_64' -nvidia_cuda_nvcc_cu12==12.5.82; platform_machine == 'x86_64' -nvidia-cuda-nvrtc-cu12==12.5.82; platform_machine == 'x86_64' -nvidia_cuda_runtime_cu12==12.5.82; platform_machine == 'x86_64' -nvidia_cusolver_cu12==11.6.3.*; platform_machine == 'x86_64' -nvidia_cusparse_cu12==12.5.1.*; platform_machine == 'x86_64' -nvidia_nccl_cu12==2.23.4; platform_machine == 'x86_64' -nvidia_nvjitlink_cu12==12.5.82; platform_machine == 'x86_64' +nvidia-cuda-cupti-cu12==12.9.79; platform_machine == 'x86_64' +nvidia-cublas-cu12==12.9.1.*; platform_machine == 'x86_64' +nvidia-cudnn-cu12==9.19.0.*; platform_machine == 'x86_64' +nvidia-cufft-cu12==11.4.1.*; platform_machine == 'x86_64' +nvidia-curand-cu12==10.3.10.*; platform_machine == 'x86_64' +nvidia-cuda-nvcc-cu12==12.9.86; platform_machine == 'x86_64' +nvidia-cuda-nvrtc-cu12==12.9.86; platform_machine == 'x86_64' +nvidia-cuda-runtime-cu12==12.9.79; platform_machine == 'x86_64' +nvidia-cusolver-cu12==11.7.5.*; platform_machine == 'x86_64' +nvidia-cusparse-cu12==12.5.10.*; platform_machine == 'x86_64' +nvidia-nccl-cu12==2.29.7; platform_machine == 'x86_64' +nvidia-nvjitlink-cu12==12.9.86; 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'