ROOG
2960ac1df5
- Updated `requirements.txt` to relax PyTorch version constraint for better GPU compatibility. - Added detailed GPU setup instructions, new device fallback options, and command examples to `README.md`. - Developed a new script `plot_model_max_x_trend.py` for visualizing training trends, generating HTML/Markdown reports.
14 lines
300 B
Plaintext
14 lines
300 B
Plaintext
# Keep torch unpinned to avoid forcing old wheels on new GPUs (e.g. RTX 50xx).
|
|
torch>=2.5.1
|
|
stable-baselines3==2.3.2
|
|
gym==0.26.2
|
|
gymnasium==0.29.1
|
|
shimmy==1.3.0
|
|
gym-super-mario-bros==7.4.0
|
|
nes-py==8.2.1
|
|
opencv-python==4.10.0.84
|
|
numpy==1.26.4
|
|
tensorboard==2.18.0
|
|
imageio==2.36.1
|
|
imageio-ffmpeg==0.5.1
|