Skip to content

Commit 0d19b18

Browse files
committed
fix: Torch Upgrade to 2.2.0.dev
1 parent bb5bf00 commit 0d19b18

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ torch.jit.save(trt_ts_module, "trt_torchscript_module.ts") # save the TRT embedd
116116
These are the following dependencies used to verify the testcases. Torch-TensorRT can work with other versions, but the tests are not guaranteed to pass.
117117

118118
- Bazel 5.2.0
119-
- Libtorch 2.1.0.dev20230703 (built with CUDA 12.1)
119+
- Libtorch 2.2.0.dev (latest nightly) (built with CUDA 12.1)
120120
- CUDA 12.1
121121
- cuDNN 8.8.1
122122
- TensorRT 8.6.1

py/requirements.txt

+2-2
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@ numpy
22
packaging
33
pybind11==2.6.2
44
--extra-index-url https://download.pytorch.org/whl/nightly/cu121
5-
torch>=2.1.0.dev,<2.2.0
5+
torch>=2.2.0.dev,<2.3.0
66
torchvision>=0.16.0.dev,<0.17.0
77
--extra-index-url https://pypi.ngc.nvidia.com
88
tensorrt==8.6.1
9-
pyyaml
9+
pyyaml

pyproject.toml

+3-3
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ requires = [
99
"typing-extensions>=4.7.0",
1010
"future>=0.18.3",
1111
"tensorrt>=8.6,<8.7",
12-
"torch >=2.1.0.dev,<2.2.0",
12+
"torch >=2.2.0.dev,<2.3.0",
1313
#"torch==2.1.0.dev20230731",
1414
"pybind11==2.6.2",
1515
"numpy",
@@ -42,7 +42,7 @@ readme = {file = "py/README.md", content-type = "text/markdown"}
4242
requires-python = ">=3.8"
4343
keywords = ["pytorch", "torch", "tensorrt", "trt", "ai", "artificial intelligence", "ml", "machine learning", "dl", "deep learning", "compiler", "dynamo", "torchscript", "inference"]
4444
dependencies = [
45-
"torch >=2.1.0.dev,<2.2.0",
45+
"torch >=2.2.0.dev,<2.3.0",
4646
#"torch==2.1.0.dev20230731",
4747
"tensorrt>=8.6,<8.7",
4848
"packaging>=23",
@@ -217,4 +217,4 @@ disallow_untyped_calls = false
217217
[[tool.mypy.overrides]]
218218
module = "torch_tensorrt.fx.*"
219219
ignore_errors = true
220-
follow_imports = "skip"
220+
follow_imports = "skip"

0 commit comments

Comments
 (0)