Tällä hetkellä Jaxissa ei ole suoraa tapaa poistaa GPU -muistin edistäminen käyttämättä ympäristömuuttujia. Suositeltu menetelmä käsittää ympäristömuuttujan `xla_python_client_prealocacacate ': n asettamisen`' false 'ennen Jaxin tuomista. Tämä voidaan tehdä joko Python-skriptissäsi käyttämällä `OS.enviriron ['xla_python_client_prealocacacate'] = 'false' tai asettamalla sen ulkoisesti ennen komentosarjan suorittamista` vienti XLA_PYTHON_CLIENT_PREALLOCATE = FALSE` (Unix-tyyppisillä järjestelmillä) tai `aseta se asetettu XLA_PYTHON_CLIENT_PREATLLOCE = FALSE` (Windowsissa).
On kuitenkin tärkeää asettaa nämä ympäristömuuttujat ennen Jaxin tuomista, koska niiden asettamisella myöhemmin ei ole vaikutusta [1] [3]. Ei ole sisäänrakennettuja JAX-toimintoja tai kokoonpanoja, joiden avulla voit poistaa käytöstä esikulaatiota käyttämättä ympäristömuuttujia.
Jos kohtaat ongelmia esikulmion poistamisessa JAX: n uudemmissa versioissa, se voi johtua virheestä ja päivitysten uusimpien dokumentaatioiden tai julkaisumuistioiden tarkistaminen on suositeltavaa [9].
Viittaukset:[1] https://github.com/jax-ml/jax/discussions/21962
[2] https://learn.microsoft.com/en-us/windows/win32/msi/environment-table
[3] https://jax.readthedocs.io/en/latest/gpu_memory_allocation.html
[4] https://docs.jax.dev/en/latest/jep/4410-omnistaging.html
[5] https://kolonist26-jax-kr.readthedocs.io/en/latest/gpu_memory_allocation.html
[6] https://developer.jboss.org/thread/215838
[7] https://crekit.science/documentation/ad_systems/
.
[9] https://github.com/jax-ml/jax/discussions/19014