Pkun
@Pkun

Are you sure you want to delete the question?

If your question is resolved, you may close it.

Leaving a resolved question undeleted may help others!

We hope you find it useful!

Google ColabでStable Diffusion web UIでBasil mixが使えずランタイムエラーってなります

解決したいこと

ここに解決したい内容を記載してください。

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper__index_select)
これはどういう意味なんでしょうか?
なぜBasil mixが使えないんでしょうか?

0

1Answer

意味としては、wrapper_index_selectのメソッドでは全ての変数を同じデバイスで扱うことを想定しているのに、2つ以上のデバイス(CPUとGPU)が見つかったよ。と、いうことなので何とか全ての変数がGPU上にある状態、もしくは扱いにすれば問題ないかと。
少し違いますが参考:https://jitaku.work/it/category/ai/machine-learning/deep-learning/pytorch/found_at_least_two_devices/

1Like

Comments

  1. @Pkun

    Questioner

    めちゃくちゃ難しいですね?
    どうすればいいか分からないです。
    コードを追記すればいいんでしょうか?

  2. @Pkun

    Questioner

    import torch
    torch.set_default_tensor_type('torch.cuda.FloatTensor')
    これ書いちゃえばいいとかすか?
    なんか basilMIXってモデルの時だけ、エラー出るんすよね?

  3. 参考サイトのコードを利用すれば全部の変数がGPU扱いになるので、それでダメなら下記のサイトのハードウェアアクセラレータがGPUになっていない等が一応考えられます。

    思いつく限り、以上でエラー文通りの問題ならどうにかなるはずですが、それでも同じエラー文が出る場合はこちらも難しいところ、というかお手上げです。

  4. 稚拙な回答で申し訳ありません。

  5. @Pkun

    Questioner

    ありがとうございます!今の使用目的で目的がえちい画像生成ですので、そこからステップアップしていろいろ模索したい所なので、凄く助かります!この技術で凄い貢献できるようになりいです♪
    アドバイスありがとうございますm(_ _)m

Your answer might help someone💌