From 4946c7a54faa983a650bd628cf7a70c654a991d1 Mon Sep 17 00:00:00 2001 From: tntwise Date: Tue, 27 Aug 2024 19:27:25 -0700 Subject: [PATCH] update windows deps --- src/DownloadDeps.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/DownloadDeps.py b/src/DownloadDeps.py index 7ca68715..920b371f 100644 --- a/src/DownloadDeps.py +++ b/src/DownloadDeps.py @@ -201,9 +201,9 @@ def getPyTorchCUDADeps(self): torchCUDAWindowsDeps = [ "https://github.com/TNTwise/real-video-enhancer-models/releases/download/models/spandrel-0.3.4-py3-none-any.whl", # "--pre", - "https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240816%2Bcu124-cp311-cp311-win_amd64.whl", + "https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240826%2Bcu124-cp311-cp311-win_amd64.whl", # "--pre", - "https://download.pytorch.org/whl/nightly/cu124/torchvision-0.20.0.dev20240817%2Bcu124-cp311-cp311-win_amd64.whl", + "https://download.pytorch.org/whl/nightly/cu124/torchvision-0.20.0.dev20240826%2Bcu124-cp311-cp311-win_amd64.whl", # "torch==2.4.0", # "torchvision==0.19.0", # "safetensors", @@ -245,15 +245,15 @@ def getTensorRTDeps(self): tensorRTDeps += ["https://download.pytorch.org/whl/nightly/cu124/torch_tensorrt-2.5.0.dev20240826%2Bcu124-cp311-cp311-linux_x86_64.whl"] case "win32": tensorRTDeps = [ - "tensorrt==10.1.0", - "tensorrt_cu12==10.1.0", - "tensorrt-cu12_libs==10.1.0", - "tensorrt_cu12_bindings==10.1.0", + "tensorrt==10.3.0", + "tensorrt_cu12==10.3.0", + "tensorrt-cu12_libs==10.3.0", + "tensorrt_cu12_bindings==10.3.0", ] tensorRTDeps += ( # "--no-deps", # "torch_tensorrt==2.4.0", - "https://download.pytorch.org/whl/nightly/cu124/torch_tensorrt-2.5.0.dev20240816%2Bcu124-cp311-cp311-win_amd64.whl", + "https://download.pytorch.org/whl/nightly/cu124/torch_tensorrt-2.5.0.dev20240826%2Bcu124-cp311-cp311-win_amd64.whl", ) return tensorRTDeps