Update README.md

marcelowds · Aug 9, 2022 · 8a05f9b · 8a05f9b
1 parent ee5965d
commit 8a05f9b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -2,6 +2,14 @@ Prepare conda environment
 
 ```conda create -n srsde python=3.8.2```
 
+Install requirements
+
+```pip3 install -r requirements.txt```
+
+Also install jax+cuda
+
+```pip install --upgrade jax==0.2.8 jaxlib==0.1.59+cuda110 -f https://storage.googleapis.com/jax-releases/jax_releases.html```
+
 Activate conda environment
 
 ```conda activate srsde_env```