RuntimeError: vk::PhysicalDevice::createDeviceUnique: ErrorInitializationFailed on A100 (when using other gpus except CUDA_VISIBLE_DEVICES=0)

[Liang-ZX]

Hi authors,

Thanks for your wonderful work. But I met a runtime error when I run

CUDA_VISIBLE_DEVICES=4 python simpler_env/simple_inference_visual_matching_prepackaged_envs.py --policy rt1 \
        --ckpt-path ./checkpoints/rt_1_tf_trained_for_000400120  --task google_robot_pick_coke_can  \
        --logging-root ./results_simple_eval/  --n-trajs 10

It only works when I run it on CUDA_VISIBLE_DEVICES=0. How can I fix it? Thank you.

The vulkaninfo is shown below.

[xuanlinli17]

Try DISPLAY="" CUDA_VISIBLE_DEVICES=x python xxx

[LukeLIN-web]

Hi authors,

Thanks for your wonderful work. But I met a runtime error when I run

CUDA_VISIBLE_DEVICES=4 python simpler_env/simple_inference_visual_matching_prepackaged_envs.py --policy rt1
--ckpt-path ./checkpoints/rt_1_tf_trained_for_000400120 --task google_robot_pick_coke_can
--logging-root ./results_simple_eval/ --n-trajs 10 It only works when I run it on CUDA_VISIBLE_DEVICES=0. How can I fix it? Thank you.

The vulkaninfo is shown below.

Did you find the solution?