JAX mälu jaotamise konfigureerimiseks võib kasutada mitmeid keskkonnamuutujaid:
1.. Selle seadistamine "Vale", keelab eelkoha kohaldamise, põhjustades JAX -i vajaduse korral mälu eraldamise. Kuid see võib põhjustada mälu killustumist ja ei pruugi Jaxi mõnes versioonis [1] [3] [5] [5] alati töötada, nagu võib oodata.
2. Näiteks selle seadistamine väärtuseks. See võib aidata vältida mäluväliseid vigu, kuid võib põhjustada oomi hoiatusi, kui neid ei hallata korralikult [1] [3] [5].
3. Ehkki selle aeglase jõudluse tõttu ei soovitata seda üldiseks kasutamiseks, võib see olla kasulik GPU mälu kasutamise silumiseks või minimeerimiseks [1] [3].
Neid muutujaid saab enne teie JAX -i rakenduse käivitamist seadistada, et kohandada selle mälu jaotuskäitumist. Kuid mõned kasutajad on teatanud Jaxi teatud versioonides nende seadetega ebajärjekindlat käitumist [5] [9].
Tsitaadid:[1] https://kolonist26-jax-kr.readthedocs.io/en/latest/gpu_memory_allocation.html
]
[3] https://jax.readthedocs.io/en/latest/gpu_memory_allocation.html
[4] https://docs.jax.dev/en/latest/installation.html
[5] https://github.com/google/jax/issues/19267
[6] https://www.maplesoft.com/support/help/maple/view.aspx?path=envvar
[7] https://docs.jax.dev/en/latest/faq.html
[8] https://github.com/google/jax/issues/1491
[9] https://github.com/google/jax/issues/19035
]