Google ColabでStable Diffusion web UIが本当に動くのか試してみました
こんにちは。電動自転車ほしいおじさんです。
最近話題のAIをつかってみました。
今回はわかりやすいところで画像生成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
とりあえず実行
サイトではこれであとは実行するのみということだったのでとりあえずは実行
5~10分くらいまち結果を確認してみる
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にインストールされているバージョンを確認
!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のサイトにアクセスしダウンロードを試みます
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時間たっても結果が返ってこないので、ちょっと時間をおいてから再確認してみます
続きはこちら
この記事が参加している募集
ありがとうございます!いただいたサポートは電動自転車の購入費に充てさせていただきます!