diff --git a/requirements.txt b/requirements.txt
index 16071648..384bbb81 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,10 +18,10 @@ dbus-python
 toml
 python-dateutil
 websockets
-torch==2.0.1; platform_machine=="aarch64"
+torch; platform_machine=="aarch64"
 torch>=2.0.1; platform_machine!="aarch64"
 
-torchvision==0.15.2; platform_machine=="aarch64"
+torchvision; platform_machine=="aarch64"
 torchvision>=0.15.2; platform_machine!="aarch64"
 
 stable_baselines3==1.8.0; platform_machine=="armv6l"