Upload 7 files
Browse files- .gitattributes +4 -0
- tensorrt_libs/__init__.py +56 -0
- tensorrt_libs/__pycache__/__init__.cpython-312.pyc +0 -0
- tensorrt_libs/__pycache__/__init__.cpython-313.pyc +0 -0
- tensorrt_libs/nvinfer_10.dll +3 -0
- tensorrt_libs/nvinfer_builder_resource_10.dll +3 -0
- tensorrt_libs/nvinfer_plugin_10.dll +3 -0
- tensorrt_libs/nvonnxparser_10.dll +3 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
tensorrt_libs/nvinfer_10.dll filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
tensorrt_libs/nvinfer_builder_resource_10.dll filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
tensorrt_libs/nvinfer_plugin_10.dll filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
tensorrt_libs/nvonnxparser_10.dll filter=lfs diff=lfs merge=lfs -text
|
tensorrt_libs/__init__.py
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#
|
| 2 |
+
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
| 3 |
+
# SPDX-License-Identifier: Apache-2.0
|
| 4 |
+
#
|
| 5 |
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
| 6 |
+
# you may not use this file except in compliance with the License.
|
| 7 |
+
# You may obtain a copy of the License at
|
| 8 |
+
#
|
| 9 |
+
# http://www.apache.org/licenses/LICENSE-2.0
|
| 10 |
+
#
|
| 11 |
+
# Unless required by applicable law or agreed to in writing, software
|
| 12 |
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
| 13 |
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| 14 |
+
# See the License for the specific language governing permissions and
|
| 15 |
+
# limitations under the License.
|
| 16 |
+
#
|
| 17 |
+
|
| 18 |
+
import ctypes
|
| 19 |
+
import glob
|
| 20 |
+
import os
|
| 21 |
+
import sys
|
| 22 |
+
|
| 23 |
+
CURDIR = os.path.realpath(os.path.dirname(__file__))
|
| 24 |
+
|
| 25 |
+
|
| 26 |
+
def try_load(library):
|
| 27 |
+
try:
|
| 28 |
+
ctypes.CDLL(library)
|
| 29 |
+
except OSError:
|
| 30 |
+
pass
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
def try_load_libs_from_dir(path):
|
| 34 |
+
for lib in glob.iglob(os.path.join(path, "*.so*")):
|
| 35 |
+
try_load(lib)
|
| 36 |
+
for lib in glob.iglob(os.path.join(path, "*.dll*")):
|
| 37 |
+
try_load(lib)
|
| 38 |
+
|
| 39 |
+
|
| 40 |
+
DEPENDENCY_PATHS = [
|
| 41 |
+
os.path.join("nvidia", "cuda_runtime"),
|
| 42 |
+
os.path.join("nvidia", "cuda_nvrtc"),
|
| 43 |
+
]
|
| 44 |
+
for dep_path in DEPENDENCY_PATHS:
|
| 45 |
+
try_load_libs_from_dir(
|
| 46 |
+
os.path.join(
|
| 47 |
+
CURDIR,
|
| 48 |
+
os.path.pardir,
|
| 49 |
+
dep_path,
|
| 50 |
+
"bin" if sys.platform.startswith("win") else "lib",
|
| 51 |
+
)
|
| 52 |
+
)
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
# Try loading all packaged libraries. This is a nop if there are no libraries packaged.
|
| 56 |
+
try_load_libs_from_dir(CURDIR)
|
tensorrt_libs/__pycache__/__init__.cpython-312.pyc
ADDED
|
Binary file (1.75 kB). View file
|
|
|
tensorrt_libs/__pycache__/__init__.cpython-313.pyc
ADDED
|
Binary file (1.79 kB). View file
|
|
|
tensorrt_libs/nvinfer_10.dll
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:62d50de6cebdd7b9f3870734c5c32f2983b625aebb02ec9742fc335210555b53
|
| 3 |
+
size 215426088
|
tensorrt_libs/nvinfer_builder_resource_10.dll
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:131b1bc6bcc23a8d074bee621551292c142027d5b7da877803cca9be4ce84332
|
| 3 |
+
size 1436593704
|
tensorrt_libs/nvinfer_plugin_10.dll
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e7c6328fa39e4077df1c13fc528b47fc9c00d07f6d1717ade7e5b4ee4b19e7d2
|
| 3 |
+
size 30986792
|
tensorrt_libs/nvonnxparser_10.dll
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d2ac4059ff7f784e37cf2c0602bf3145d3d9fe767d43c32f9b6be2cd4a25695d
|
| 3 |
+
size 3064872
|