tutoriais:jax
Differences
This shows you the differences between two versions of the page.
| Next revision | Previous revision | ||
| tutoriais:jax [2024-05-13 14:47] – created 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 12: | Line 12: | ||
| </ | </ | ||
| - | A vantagem de usar o apptainer é que o ambiente é bastante controlado e há suporte nativo à GPU. | + | A vantagem de usar o apptainer é que o ambiente é bastante controlado e há suporte nativo à GPU Nvidia. |
| Opção 2: Usar o conda: | Opção 2: Usar o conda: | ||
| Line 18: | Line 18: | ||
| # criar o ambiente | # criar o ambiente | ||
| conda create -n jax | conda create -n jax | ||
| + | conda init bash | ||
| + | source .bashrc | ||
| + | conda activate jax | ||
| conda install pip | conda install pip | ||
| Line 23: | Line 26: | ||
| pip install jax[cpu] | pip install jax[cpu] | ||
| - | # ou | + | # se houver GPU da Nvidia no computador usar jax[cuda12] em vez de jax[cpu] |
| - | pip install | + | |
| </ | </ | ||
| A vantagem de usar o conda é usar o procedimento de instalação descrito na documentação oficial. | A vantagem de usar o conda é usar o procedimento de instalação descrito na documentação oficial. | ||
| - | | + | |
| + | https:// | ||
tutoriais/jax.1715622460.txt.gz · Last modified: by gnann