eightban's memo

残しておきたい記事をまとめてみました。このブログに書いてあるドキュメントやブログで配布しているファイルの使用によって発生するいかなる損害に対してもこのブログの管理者は責任を負いません。使用する場合は自己責任のもとに使用してください。

stable-diffusion

Stable Diffusion (Diffusers) / DiffusersでcontrolnetをCPUだけで動かす

投稿日:

GPU固定でプログラムをされているとCPUでは動きません

/usr/local/lib/python3.10/dist-packages/torch/cuda/init.py in _lazy_init()
245 if 'CUDA_MODULE_LOADING' not in os.environ:
246 os.environ['CUDA_MODULE_LOADING'] = 'LAZY'
--> 247 torch._C._cuda_init()
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver

pipeline_controlnet.py

/usr/local/lib/python3.10/dist-packages/diffusers/pipelines/controlnet/pipeline_controlnet.py

次のコードに命令を追加して書き換えます


        device = torch.device(f"cuda:{gpu_id}")
        device = "cuda" if torch.cuda.is_available() else "cpu"

-stable-diffusion

Copyright© eightban's memo , 2024 All Rights Reserved Powered by STINGER.