diff --git a/runai_torch_requirements.txt b/runai_torch_requirements.txt
index abe14c9..909023f 100644
--- a/runai_torch_requirements.txt
+++ b/runai_torch_requirements.txt
@@ -1,4 +1,4 @@
 --index-url https://download.pytorch.org/whl/cu121
 torch==2.2.1
-torchvision==0.17.1
+torchvision==0.19.1+cu118
 torchaudio==2.2.1
\ No newline at end of file
diff --git a/torch_requirements.txt b/torch_requirements.txt
index 489c1ca..5cbc327 100644
--- a/torch_requirements.txt
+++ b/torch_requirements.txt
@@ -1,4 +1,4 @@
 --index-url https://download.pytorch.org/whl/cu118
 torch==2.0.0
-torchvision==0.15.1
+torchvision==0.19.1+cu118
 torchaudio==2.0.1
\ No newline at end of file