add h5py in requirement.txt and add conda intall atari in readme

ikostrikov · Nov 24, 2021 · c2925c9 · c2925c9
1 parent 6bc954d
commit c2925c9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git 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
@@ -1,4 +1,5 @@
 gym
 matplotlib
 pybullet
-stable-baselines3
+stable-baselines3
+h5py