GPUのメモリが2Gしかないので下記エラーとなる
torch.cuda.OutOfMemoryError: CUDA out of memory
GPU: GeForce GTX 1050 ( メモリ2G , Compute Capability = 6.1 , Pascal世代 )
CPU: 4コア
Memory: 32G
Disk: 300G
OS: Ubuntu 22.04
CUDA: 12.4.1
Python: 3.10.12
参考URL:
https://kb.seeck.jp/archives/23226
https://note.com/gladevise/n/ne2b2bac5969c
https://qiita.com/mml/items/16a8618debbc76e97fd0
https://ameblo.jp/memescat/entry-12867374685.html
https://qiita.com/tsuyuni/items/98ed25457488ede19a48
https://www.nvidia.com/ja-jp/drivers/
https://pytorch.org/
https://developer.nvidia.com/cuda-toolkit-archive
-- 1. 初期設定
sudo su -
update-alternatives --config editor
visudo
sudo apt update
sudo apt upgrade
sudo apt install wget git python3 python3-venv
python3 --version
sudo apt install python3-pip
df -h
sudo lvextend -r -l 100%VG /dev/ubuntu-vg/ubuntu-lv
df -h
-- 2. ドライバインストール
https://www.nvidia.com/ja-jp/drivers/
Linux x64 (AMD64/EM64T) Display Driver 570.144 | Linux 64-bit
chmod +x NVIDIA-Linux-x86_64-570.144.run
sudo ./NVIDIA-Linux-x86_64-570.144.run
sudo reboot
nvidia-smi
-- 3. CUDAインストール
https://developer.nvidia.com/cuda-toolkit-archive
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda-repo-ubuntu2204-12-4-local_12.4.1-550.54.15-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-4-local_12.4.1-550.54.15-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-4
/usr/local/cuda-12.4/bin/nvcc --version
vi ~/.bashrc
# Setting CUDA PATH
export PATH="/usr/local/cuda-12.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH"
. ~/.bashrc
which nvcc
nvcc --version
-- 4. PyTorchインストールコマンドの確認
https://pytorch.org/
-- pip3 install torch torchvision torchaudio
pip3 -> pipに置き換えてwebui-user.shの「TORCH_COMMAND」に指定する
-- 5. Stable Diffusion WebUIインストール
git clone --depth=1 --branch v1.2.1 https://github.com/AUTOMATIC1111/stable-diffusion-webui
cd stable-diffusion-webui
vi webui-user.sh
export TORCH_COMMAND="pip install torch torchvision torchaudio"
export COMMANDLINE_ARGS="--autolaunch --xformers --listen"
※ COMMANDLINE_ARGS末尾に--listen追加
chmod 755 webui-user.sh
bash ./webui-user.sh
vi requirements_versions.txt
httpx==0.24.1
※ 末尾httpx追加
bash ./webui.sh
