Blackwellで sd-scritpsを実行する場合、次の組み合わせで動作確認できました。
python -m pip install --upgrade torch==2.11.0 torchvision==0.26.0 torchaudio==2.11.0 --index-url https://download.pytorch.org/whl/cu128
python -m pip install --upgrade --no-deps bitsandbytes
pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
python -m pip install -U --no-deps bitsandbytes
pip install torch==2.7.0+cu128 torchvision==0.22.0+cu128 --index-url https://download.pytorch.org/whl/cu128
pip install xformers==0.0.30
pip install -U -r requirements.txt
バージョン指定しない場合のコマンド
pip install -U pip setuptools wheel
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
python -m pip install -U --no-deps bitsandbytes
インストール後次のコマンドを実行します。
python -c "import torch; print(torch.__version__); print(torch.version.cuda); print(torch.cuda.get_device_name(0)); print(torch.cuda.get_device_capability(0)); print(torch.cuda.get_arch_list())"
以下の形式でコマンド結果が表示されます。sm_120 が表示されることが確認できればOKです。
2.11.0+cu128
12.8
NVIDIA GeForce RTX 5090
(12, 0)
['sm_75', 'sm_80', 'sm_86', 'sm_90', 'sm_100', 'sm_120']
