Skip to content

Commit

Permalink
9.3.0 Upgrade and version fixing
Browse files Browse the repository at this point in the history
  • Loading branch information
gs-olive committed Mar 22, 2024
1 parent 0779156 commit 1e418b6
Show file tree
Hide file tree
Showing 4 changed files with 15 additions and 4 deletions.
8 changes: 8 additions & 0 deletions packaging/pre_build_script_windows.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
python -m pip install pyyaml packaging numpy
python -c "import torch; print('IMPORTED TORCH 1')"
python -m pip config set global.index-url "https://pypi.org/simple https://download.pytorch.org/whl/nightly/cu118 https://download.pytorch.org/whl/nightly/cu121"
python -m pip config set global.extra-index-url "https://pypi.nvidia.com"
python -m pip config set global.no-cache-dir true
python -c "import torch; print('IMPORTED TORCH 2')"
python -m pip install tensorrt==9.3.0.post12.dev1 tensorrt_libs==9.3.0.post12.dev1 tensorrt_bindings==9.3.0.post12.dev1
python -c "import torch; print('IMPORTED TORCH 3')"
3 changes: 3 additions & 0 deletions packaging/smoke_test_windows.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
import torch
import tensorrt
import torch_tensorrt
4 changes: 2 additions & 2 deletions py/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,6 @@ pybind11==2.6.2
--extra-index-url https://download.pytorch.org/whl/nightly/cu121
torch>=2.4.0.dev,<2.5.0
torchvision>=0.18.0.dev,<0.19.0
--extra-index-url https://pypi.ngc.nvidia.com
tensorrt==8.6.1
--extra-index-url https://pypi.nvidia.com
tensorrt==9.3.0.post12.dev1
pyyaml
4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ requires = [
"cffi>=1.15.1",
"typing-extensions>=4.7.0",
"future>=0.18.3",
"tensorrt>=8.6,<8.7",
"tensorrt==9.3.0.post12.dev1",
"torch >=2.4.0.dev,<2.5.0",
"pybind11==2.6.2",
"numpy",
Expand Down Expand Up @@ -42,7 +42,7 @@ requires-python = ">=3.8"
keywords = ["pytorch", "torch", "tensorrt", "trt", "ai", "artificial intelligence", "ml", "machine learning", "dl", "deep learning", "compiler", "dynamo", "torchscript", "inference"]
dependencies = [
"torch >=2.4.0.dev,<2.5.0",
"tensorrt>=8.6,<8.7",
"tensorrt==9.3.0.post12.dev1",
"packaging>=23",
"numpy",
"typing-extensions>=4.7.0",
Expand Down

0 comments on commit 1e418b6

Please sign in to comment.