Google ColabでStable Diffusion web UIが本当に動くのか試してみました

こんにちは。電動自転車ほしいおじさんです。

最近話題のAIをつかってみました。

今回はわかりやすいところで画像生成AIを使ってみます。

どうやって使うのかなって色々試していると以下の記事にGoogle ColabでStable Diffusion web UIを使ってやるっていうのを見つけたので、簡単にできそうだったのでやってみました。

【AI画像生成】Google ColabでStable Diffusion web UIを動かす方法

Google Colabにログイン

まずはGoogle Colabにログインを行い新規ワークブックでTEST用のワークブックを作成しました

コードをGoogle Colabに書き込んでみる

今回記載したコードはサイトにもある以下をそのまま引用しました。

!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
%cd /content/stable-diffusion-webui

!wget https://huggingface.co/nuigurumi/basil_mix/resolve/main/Basil_mix_fixed.safetensors -O /content/stable-diffusion-webui/models/Stable-diffusion/Basil_mix_fixed.safetensors
!wget https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors -O /content/stable-diffusion-webui/models/VAE/vae-ft-mse-840000-ema-pruned.safetensors

!python launch.py --share --xformers --enable-insecure-extension-access

とりあえず実行

サイトではこれであとは実行するのみということだったのでとりあえずは実行

５～１０分くらいまち結果を確認してみる
Running on public URL のような表記がでればOKのようだが結果は如何に・・

なんと！エラーがでてました

RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=11.7 and torchvision has CUDA Version=11.8. Please reinstall the torchvision that matches your PyTorch install.

調べてみるとどうやらバージョン差異があるようなので揃えてあげないといけません

既存のバージョンを確認

Google ColabでCUDAとtorchのバージョンを変更する。 - Qiita

上記サイトをもとにGoogle Colabにインストールされているバージョンを確認

!nvcc --version

!nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0

!python -c 'import torch; print(torch.__version__) '

!python -c 'import torch; print(torch.__version__) '

2.0.0+cu117

とりあえずPyTourのサイトにアクセスしダウンロードを試みます

Start Locally

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

このままじゃ構文エラーとなるのでpip3部分を!pipに変更する

!pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

これでうまくいくかわからないけど試してみます

Successfully installed torchaudio-2.0.1+cu118

とりあえずインストールは成功したようなので、もう1度レッツトライ！！

・・・といいつも1時間たっても結果が返ってこないので、ちょっと時間をおいてから再確認してみます

続きはこちら