Can I ignore warnings for jaxlib and TPU? #22434
Answered
by
Lime-Cakes
ToshiyukiBandai
asked this question in
Q&A
-
Hi all, JAX gives me the following two warnings. I get them when running on both CPU and GPU. Can I safely ignore them? Or, do they indicate some issues in my GPU or performance degradation? In fact,
System InformationEnvironment: WSL2 nvidia-smi
|
Beta Was this translation helpful? Give feedback.
Answered by
Lime-Cakes
Jul 14, 2024
Replies: 1 comment 1 reply
-
It's safe to ignore. rocm is for amd device, and tpu is for google's cloud tpu, neither relevant with nvidia device. However, segmentation fault in nvidia-smi might point to an issue with nvidia software, probably unrelated to jax. |
Beta Was this translation helpful? Give feedback.
1 reply
Answer selected by
ToshiyukiBandai
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
It's safe to ignore. rocm is for amd device, and tpu is for google's cloud tpu, neither relevant with nvidia device.
However, segmentation fault in nvidia-smi might point to an issue with nvidia software, probably unrelated to jax.