jax问题(在NVDIA DGX机器上，同样如此)

Question

我试图在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)

任何建议都将不胜感激。

Answer 1

这意味着您的CUDA安装配置不正确，通常可以通过确保$PATH中存在CUDA工具包二进制文件(包括ptxas)来修复。有关对报告类似问题的用户的响应，请参见https://github.com/google/jax/discussions/6843和https://github.com/google/jax/issues/7239。