external/local_xla/xla/tsl/cuda/cudart_stub.cc:32] Could not find cuda drivers on your machine, GPU will not be used.
Opened this issue · 0 comments
bc129697 commented
when i run "CUDA_VISIBLE_DEVICES=0 python pipeline_i2v_eval_v01.py
--denoise_checkpoint "ckpts/first_stage.pt"
--image_path "demo/3.png"
--output_dir "outputs/3""
this error "external/local_xla/xla/tsl/cuda/cudart_stub.cc:32] Could not find cuda drivers on your machine, GPU will not be used.", how to solve this problem?thanks