我試圖在 nvidia dgx 機器上運行 jax，但失敗得很慘，因此：
>>> import jax
>>> import jax.numpy as jnp
>>> x = jnp.arange(10)
2021-10-25 13:00:05.863667: W
external/org_tensorflow/tensorflow/stream_executor/gpu/asm_compiler.cc:80] Couldn't
get ptxas version string: INTERNAL: Couldn't invoke ptxas --version
2021-10-25 13:00:05.864713: F
external/org_tensorflow/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc:435]
ptxas returned an error during compilation of ptx to sass: 'INTERNAL: Failed to
launch ptxas' If the error message indicates that a file could not be written,
please verify that sufficient filesystem space is provided.
Aborted (core dumped)
任何建議將不勝感激。
uj5u.com熱心網友回復：
這意味著您的CUDA安裝不正確配置，并且通常可以通過確保CUDA工具包二進制檔案（包括被固定ptxas）存在于您的$PATH。請參閱https://github.com/google/jax/discussions/6843和https://github.com/google/jax/issues/7239以了解對報告類似問題的用戶的回應。
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/337538.html