Skip to content

Commit fca65b4

Browse files
committed
PEP517 installation of torchTRT
1 parent 7b3bc6e commit fca65b4

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ requires = [
88
"cffi>=1.15.1",
99
"typing-extensions>=4.7.0",
1010
"future>=0.18.3",
11-
"torch @ file:////usr/local/lib/python3.12/dist-packages/torch"
11+
"torch>=2.6.0.dev,<2.7.0",
1212
"pybind11==2.6.2",
1313
"numpy",
1414
]
@@ -52,7 +52,7 @@ keywords = [
5252
"inference",
5353
]
5454
dependencies = [
55-
"torch @ file:////usr/local/lib/python3.12/dist-packages/torch"
55+
"torch>=2.6.0.dev,<2.7.0",
5656
#"tensorrt-cu12>=10.6.0,<10.8.0",
5757
#"tensorrt-cu12-bindings>=10.6.0,<10.8.0",
5858
#"tensorrt-cu12-libs>=10.6.0,<10.8.0",

setup.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -119,10 +119,8 @@ def load_dep_info():
119119
gpu_arch_version = f"cu{__cuda_version__.replace('.','')}"
120120

121121

122-
if RELEASE:
123-
__version__ = os.environ.get("BUILD_VERSION")
124-
else:
125-
__version__ = f"{get_base_version()}.dev0+{get_git_revision_short_hash()}"
122+
123+
__version__ = os.environ.get("BUILD_VERSION")
126124

127125
if "--ci" in sys.argv:
128126
sys.argv.remove("--ci")

0 commit comments

Comments
 (0)