Merge pull request #285 from lijiyao919/master

add h5py in requirement.txt and add conda intall atari in readme
ikostrikov · Nov 24, 2021 · 41332b7 · 41332b7
2 parents 6bc954d + c2925c9
commit 41332b7
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