h1hand-walk-v0 Environment Test Failure: AttributeError in jax_cuda12_plugin._triton

[ExuberantWitness]

Description:
When attempting to test the h1hand-walk-v0 environment using the command python -m humanoid_bench.test_env --env h1hand-walk-v0, an AttributeError: module 'jax_cuda12_plugin._triton' has no attribute 'get_arch_details' is encountered.

Reproduction Steps:
Activate the humanoidbench conda environment.

Navigate to the humanoid-bench-main project directory.

Execute the command: python -m humanoid_bench.test_env --env h1hand-walk-v0

Expected Behavior:
The environment should load successfully and the test should run without errors.

Actual Behavior:
The program raises an AttributeError: module 'jax_cuda12_plugin._triton' has no attribute 'get_arch_details' and terminates.

Error Stack Trace:
pgsql
复制
(humanoidbench) exuber@exuber-System-Product-Name:~/CODE/pythonProject/humanoid-bench-main$ python -m humanoid_bench.test_env --env h1hand-walk-v0
Traceback (most recent call last):
File "", line 189, in _run_module_as_main
File "", line 112, in _get_module_details
File "/home/exuber/CODE/pythonProject/humanoid-bench-main/humanoid_bench/init.py", line 3, in
from .env import ROBOTS, TASKS
File "/home/exuber/CODE/pythonProject/humanoid-bench-main/humanoid_bench/env.py", line 18, in
from .wrappers import (
File "/home/exuber/CODE/pythonProject/humanoid-bench-main/humanoid_bench/wrappers.py", line 5, in
from humanoid_bench.mjx.flax_to_torch import TorchModel, TorchPolicy
File "/home/exuber/CODE/pythonProject/humanoid-bench-main/humanoid_bench/mjx/flax_to_torch.py", line 3, in
import jax.numpy as jnp
File "/home/exuber/anaconda3/envs/humanoidbench/lib/python3.11/site-packages/jax/init.py", line 25, in
from jax._src.cloud_tpu_init import cloud_tpu_init as _cloud_tpu_init
File "/home/exuber/anaconda3/envs/humanoidbench/lib/python3.11/site-packages/jax/_src/cloud_tpu_init.py", line 17, in
from jax._src import config
File "/home/exuber/anaconda3/envs/humanoidbench/lib/python3.11/site-packages/jax/_src/config.py", line 27, in
from jax._src import lib
File "/home/exuber/anaconda3/envs/humanoidbench/lib/python3.11/site-packages/jax/_src/lib/init.py", line 121, in
import jaxlib.gpu_triton as gpu_triton # pytype: disable=import-error
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/exuber/anaconda3/envs/humanoidbench/lib/python3.11/site-packages/jaxlib/gpu_triton.py", line 38, in
get_arch_details = _cuda_triton.get_arch_details
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: module 'jax_cuda12_plugin._triton' has no attribute 'get_arch_details'
Environment Information:
Operating System: [Your OS, e.g., Ubuntu 22.04]

Python Version: Python 3.11 (from conda environment)