Praegu ei ole otsest viisi GPU mälu eelkoolimise keelamiseks Jaxis ilma keskkonnamuutujaid kasutamata. Soovitatav meetod hõlmab keskkonnamuutuja `xla_python_client_prealLocate'i seadistamist enne Jaxi importimist` 'false' '. Seda saab teha kas teie pythoni skripti raames, kasutades `os.environ ['xla_python_client_prealLocate'] = 'false' 'või seadistades selle väliselt enne skripti käivitamist` ekspordiga xla_python_client_preallocate = false ’(unix-sarnased süsteemid) või` komplektis) või `komplektis) või` komplektis) Xla_python_client_preallocate = false` (Windowsis).
Enne Jaxi importimist on aga nende keskkonnamuutujate seadistamine ülioluline, kuna nende hiljem seadistamine ei mõjuta [1] [3]. Puuduvad sisseehitatud JAX-i funktsioonid ega konfiguratsioonid, mis võimaldaksid teil eelneva lokatsiooni keelata ilma keskkonnamuutujaid kasutamata.
Kui teil on Jaxi uuemates versioonides esialgse osutamise keelamise probleeme, võib see olla tingitud veast ja värskenduste uusimate dokumentide või väljalaskemärkuste kontrollimine on soovitatav [9].
Tsitaadid:[1] https://github.com/jax-ml/jax/dicussions/21962
[2] https://learn.microsoft.com/en-us/windows/win32/mi
[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://crikit.science/documentation/ad_systems/
]
[9] https://github.com/jax-ml/jax/Discussions/19014