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

前回の記事

再度実行してみた結果

あれから時間をおいて再度実行してみましたが、結果はダメでした
こういう場合はいったん１つ１つ問題を解消してみることにしましょう

torchvisionのバージョン変更

まず今回はtorchvisionとPyTorchにて使われているCUDAのバージョンが相違があるのが根本原因のようです。

なのでまずはどちらかのバージョンをそろえたほうがいいのかな

PyTorchは11.7でtorchvisionは11.8ということなので一旦はtorchvisionのバージョンを下げようと思います

しかしCUDA 11.7に対応するバージョンのtorchvisionは存在しないということが判明！

なくなくtorchvisionのバージョンアップはあきらめました

PyTorchのバージョン変更

こちらは一応できそうっぽいのでPyTorchのバージョンを11.8に変更しようと思います

まずインストールを実施

!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-11-8_11.8.0-1_amd64.deb
!dpkg -i cuda-11-8_11.8.0-1_amd64.deb
!apt-get update

そのあとにパスの変更を実施

import os
os.environ['PATH'] += ':/usr/local/cuda-11.8/bin'
os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-11.8/lib64'

これでうまくいくことを祈っておきますがGPUが足りなくなったりとで中々うまくいかないです。

ちょっと時間あけてのんびりと試していこうと思いますがColabだと中々厳しいんだろうなという印象です。

Stable Diffusion web UI　を使うのであればGPU搭載のパソコンを用意するほうが個人的には良いなという。色々試した結果です

続きはこちら