Jax (library): Difference between revisions
| Line 2: | Line 2: | ||
==Install== | ==Install== | ||
===Conda=== | |||
<pre> | <pre> | ||
conda install jax jaxlib=*=*cuda* -c conda-forge | conda install jax jaxlib=*=*cuda* -c conda-forge | ||
</pre> | |||
===pip=== | |||
<pre> | |||
pip install --upgrade jax jaxlib==0.3.10+cuda11.cudnn82 -f https://storage.googleapis.com/jax-releases/jax_releases.html | |||
</pre> | </pre> | ||
Latest revision as of 00:35, 29 May 2022
Notes about Jax
Install
Conda
conda install jax jaxlib=*=*cuda* -c conda-forge
pip
pip install --upgrade jax jaxlib==0.3.10+cuda11.cudnn82 -f https://storage.googleapis.com/jax-releases/jax_releases.html