external/local_xla/xla/tsl/cuda/cudart_stub.cc:32] Could not find cuda drivers on your machine, GPU will not be used.

yanghb22-fdu / Hi3D-Official

[MM24] Official codes and datasets for ACM MM24 paper "Hi3D: Pursuing High-Resolution Image-to-3D Generation with Video Diffusion Models".

MIT License

215 stars 12 forks source link

external/local_xla/xla/tsl/cuda/cudart_stub.cc:32] Could not find cuda drivers on your machine, GPU will not be used. #11

Open bc129697 opened 2 months ago

bc129697 commented 2 months ago

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