diff --git a/README.md b/README.md
index 5784361b1..365555890 100644
--- a/README.md
+++ b/README.md
@@ -53,6 +53,9 @@ conda install pytorch torchvision -c soumith
 
 # Other requirements
 pip install -r requirements.txt
+
+# Gym Atari
+conda install -c conda-forge gym-atari
 ```
 
 ## Contributions
diff --git a/requirements.txt b/requirements.txt
index b996ad420..ed4d617e2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 gym
 matplotlib
 pybullet
-stable-baselines3
\ No newline at end of file
+stable-baselines3
+h5py
\ No newline at end of file