diff --git a/amd_requirements.txt b/amd_requirements.txt index 940bb6c..0bb841c 100644 --- a/amd_requirements.txt +++ b/amd_requirements.txt @@ -1,6 +1,6 @@ # For AMD GPUs with DirectML (Windows) torch==2.5.1 -torchvision==0.20.0 +torchvision==0.20.1 torchaudio==2.5.1 onnx==1.17.0 onnxruntime-directml==1.19.2 diff --git a/nvidia_requirements.txt b/nvidia_requirements.txt index 067540c..56324f4 100644 --- a/nvidia_requirements.txt +++ b/nvidia_requirements.txt @@ -1,6 +1,6 @@ # For NVIDIA GPUs with CUDA support (Windows) torch==2.5.1+cu118 -torchvision==0.20.1+cu118 +torchvision==0.20.1 torchaudio==2.5.1+cu118 onnx==1.17.0 onnxruntime-gpu==1.19.2