Home Arrow Icon Knowledge base Arrow Icon Global Arrow Icon Millised on keskkonnamuutujad, mida kasutatakse JAX mälu jaotamise konfigureerimiseks


Millised on keskkonnamuutujad, mida kasutatakse JAX mälu jaotamise konfigureerimiseks


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
]