tutoriais:jax
Differences
This shows you the differences between two versions of the page.
| Both sides previous revisionPrevious revisionNext revision | Previous revision | ||
| tutoriais:jax [2024-05-13 14:50] – gnann | tutoriais:jax [2024-05-13 15:43] (current) – gnann | ||
|---|---|---|---|
| Line 1: | Line 1: | ||
| ===== JAX ===== | ===== JAX ===== | ||
| - | Como usar o JAX nas máquinas da Rede IME? | + | Como usar o JAX nas máquinas da Rede IME. |
| Opção 1: Usar apptainer: | Opção 1: Usar apptainer: | ||
| Line 17: | Line 17: | ||
| <code bash> | <code bash> | ||
| # criar o ambiente | # criar o ambiente | ||
| - | export PATH=$PATH:/ | ||
| conda create -n jax | conda create -n jax | ||
| + | conda init bash | ||
| + | source .bashrc | ||
| conda activate jax | conda activate jax | ||
| conda install pip | conda install pip | ||
| Line 25: | Line 26: | ||
| pip install jax[cpu] | pip install jax[cpu] | ||
| - | # se houver GPU da Nvidia no computador | + | # se houver GPU da Nvidia no computador |
| - | pip install | + | |
| </ | </ | ||
tutoriais/jax.1715622615.txt.gz · Last modified: 2024-05-13 14:50 by gnann