Trenutno ni neposrednega načina za onemogočanje predhodnika GPU pomnilnika v Jaxu brez uporabe spremenljivk okolja. Priporočena metoda vključuje nastavitev spremenljivke okolja `xla_python_client_preallocate` to` 'false'`, preden uvozi Jax. To lahko storite bodisi v vašem skriptu Python z uporabo `Os.environ ['xla_python_client_preallocate'] = 'false'`` ali tako, da ga nastavite zunaj, preden zaženete svoj skript z `izvozi xla_python_client_prealLocate = false` (na sistemih, ki so podobni Unix) ali` Xla_python_client_preallocate = false` (v sistemu Windows).
Vendar je ključnega pomena, da nastavite te spremenljivke okolja pred uvozom JAX, saj jih nastavite pozneje [1] [3]. Ni vgrajenih funkcij JAX ali konfiguracij, ki bi omogočale onemogočanje predreallokacije brez uporabe spremenljivk okolja.
Če naletite na težave z onemogočanjem predreallokacije v novejših različicah Jaxa, je to lahko posledica napake in preverjanje najnovejše dokumentacije ali opomb o izdaji za posodobitve je priporočljivo [9].
Navedbe:[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/sl/latest/gpu_memory_allocation.html
[4] https://docs.jax.dev/sl/latest/jep/4410-omnistaging.html
[5] https://kolonist26-jax-kr.readthedocs.io/sl/latest/gpu_memory_allocation.html
[6] https://developer.jboss.org/thread/215838
[7] https://crikit.science/documentation/ad_systems/
[8] https://docs.nvidia.com/deeplearning/dali/user-guide/docs/advanced_topics_performance_tuning.html
[9] https://github.com/jax-ml/jax/discussions/19014