Eureka! Turns out rocm driver versions weren't the issue. sb3 above this version now attempts to use torch.load weights_only=True related to DLR-RM/stable-baselines3#1852 😮‍💨

Author: mscs

requirements.txt

@@ -5,14 +5,13 @@ numba
 numpy
 pandas
 Pillow
-pyboy==2.0.0
+pyboy>=2.0.0
 scipy
 Shimmy
-stable-baselines3
+stable-baselines3[extra]==2.2.1
 tensorboard
 tensorflow
 timg
 torch
-torchaudio
 torchvision
 tqdm