Update README.md

Fixed craftax installation instructions
RLE-Foundation · Mar 22, 2024 · 44b411d · 44b411d
1 parent ef50681
commit 44b411d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rllte/env/README.md b/rllte/env/README.md
@@ -18,7 +18,7 @@ pip install ale-py==0.8.1
 You will need a Jax GPU-enabled conda environment:
 
 ```
-conda create -n rllte jaxlib==*cuda jax python=3.11 -c conda-forge
+conda create -n craftax jaxlib=*=*cuda* jax python=3.11 -c conda-forge
 pip install craftax
 pip install brax
 pip install -e .[envs]
@@ -91,4 +91,4 @@ agent = PPO(
 agent.train(
     num_train_steps=10_000_000,
 )
-```
+```