Şu anda, JAX'ta GPU bellek ön planını çevre değişkenleri kullanmadan devre dışı bırakmanın doğrudan bir yolu yoktur. Önerilen yöntem, JAX'ı içe aktarmadan önce `` xla_python_client_preallocate` ortam değişkeninin `` yanlış '' olarak ayarlanmasını içerir. Bu, `os.environ ['xla_python_client_preallocate'] = 'yanlış' 'kullanılarak python komut dosyası içinde yapılabilir veya` `betiğinizi XLA_PYTHON_Client_PrealLocate = false (unix-like sistemlerde) veya set ile dışa aktarmadan önce harici olarak ayarlayarak yapılabilir. Xla_python_client_preallocate = false` (pencerelerde).
Bununla birlikte, JAX'ı içe aktarmadan önce bu çevre değişkenlerini ayarlamak çok önemlidir, çünkü onları daha sonra ayarlamanın bir etkisi olmayacaktır [1] [3]. Çevre değişkenlerini kullanmadan önceden öncelik devredenizi devre dışı bırakmanızı sağlayan yerleşik JAX işlevleri veya konfigürasyonları yoktur.
JAX'in yeni sürümlerinde önlemi devre dışı bırakma sorunlarıyla karşılaşırsanız, bunun bir hatadan kaynaklanıyor olabilir ve güncellemeler için en son belgelerin veya sürüm notlarının kontrol edilmesi tavsiye edilir [9].
Alıntılar:[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://deceloper.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