diff --git a/README.md b/README.md index 9feafc1..06f91e5 100644 --- a/README.md +++ b/README.md @@ -60,6 +60,7 @@ This codebase is supported on Ubuntu and has not been tested on Windows or macOS Install the requirements with: ``` conda create -n lwm python=3.10 +conda activate lwm pip install -U "jax[cuda12_pip]==0.4.23" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html pip install -r requirements.txt ```