ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Format readme

Browse Source
This commit is contained in:
rohskopf
2023-05-26 11:21:13 -06:00
parent 87850f31ca
commit 2e68930f2a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/mliap/jax/README.md
View File

@ -33,11 +33,11 @@ Use same Python dependencies as above, with some extra changes:
1. Make sure you install cupy properly! E.g.
python -m pip install cupy-cuda12x
python -m pip install cupy-cuda12x
2. Install JAX for GPU/CUDA:
python -m pip install --trusted-host storage.googleapis.com --upgrade "jax[cuda12_local]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
python -m pip install --trusted-host storage.googleapis.com --upgrade "jax[cuda12_local]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
3. Install cudNN: https://developer.nvidia.com/cudnn