diff --git a/requirements.txt b/requirements.txt
index decdc195..32e9a958 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,7 +18,7 @@ dbus-python
 toml
 python-dateutil
 websockets
-torch
-torchvision
+torch==2.0.1
+torchvision==0.15.2
 stable_baselines3
 RPi.GPIO