WSL2+Ubuntu22.04でStableSRをインストールした記録

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

nvidia-smi
Sat Aug 23 21:02:20 2025
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 575.64.01              Driver Version: 576.80         CUDA Version: 12.9     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA GeForce RTX 4070        On  |   00000000:06:00.0  On |                  N/A |
|  0%   39C    P8              5W /  200W |    1092MiB /  12282MiB |      2%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI              PID   Type   Process name                        GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|  No running processes found                                                             |
+-----------------------------------------------------------------------------------------+

git clone https://github.com/IceClear/StableSR.git
cd StableSR
python3 -m venv venv
source venv/bin/activate

torch==1.13.1 --extra-index-url https://download.pytorch.org/whl/cu118
torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cu118
albumentations==1.3.0
opencv-python==4.6.0.66
imageio==2.9.0
numpy==1.23.1
imageio-ffmpeg==0.4.2
pytorch-lightning==1.4.2
omegaconf==2.1.1
test-tube>=0.7.5
streamlit==1.12.1
einops==0.3.0
transformers==4.19.2
webdataset==0.2.5
kornia==0.6
open_clip_torch==2.0.2
invisible-watermark>=0.1.5
streamlit-drawable-canvas==0.8.0
torchmetrics==0.6.0
xformers
triton
matplotlib
wandb
pillow
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
-e git+https://github.com/openai/CLIP.git@main#egg=clip
-e .

pip install -r requirements.txt

ls weights
stablesr_000117.ckpt  stablesr_768v_000139.ckpt  stablesr_turbo.ckpt  vqgan_cfw_00011.ckpt  webui_768v_139.safetensors

#!/bin/bash
cd ~
venvFile=`pwd`'/git/StableSR/venv/bin/activate'
dataRoot='/mnt/f'
inputDir=$dataRoot'/input'
outputDir=$dataRoot'/output'
logfile=$dataRoot'/logs/StableSR.log'

cd $outputDir
if [ `ls -U1 | wc -l` -ne 0 ]; then
    rm -rf *
fi;
cd $inputDir
if [ `ls -U1 | wc -l` -eq 0 ]; then
    echo 'Not File';
    exit 1
fi;
cd ~/git/StableSR
source $venvFile
echo '*** StableSR Start ***' >> $logfile
date >> $logfile
PYTHONPATH=$PWD:$PYTHONPATH \
python scripts/sr_val_ddpm_text_T_vqganfin_oldcanvas_tile.py \
        --config configs/stableSRNew/v2-finetune_text_T_512.yaml \
        --ckpt weights/stablesr_turbo.ckpt \
        --vqgan_ckpt weights/vqgan_cfw_00011.ckpt \
        --init-img $inputDir --outdir $outputDir \
        --ddpm_steps 4 --dec_w 0.5 --seed 42 --n_samples 1 \
        --colorfix_type wavelet --upscale 4 \
        --vqgantile_size 768 --vqgantile_stride 512 --tile_overlap 48
date >> $logfile
echo '*** StableSR End ***' >> $logfile
deactivate
cd $inputDir
if [ `ls -U1 | wc -l` -ne 0 ]; then
    rm *
fi;