Open1

GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

John.K.HappyJohn.K.Happy

PyTorchのバージョンがCUDAのバージョンと合って無いときに下記のエラーが出るようだ。

GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

▼対処方法は下記のブログから知ることが出来た
https://jskangaroo.hatenablog.com/entry/2022/01/13/181658

▼pipもしくはcondaの環境ではcondaでCUDAバージョンに合致するPyTorchをインストールする。
https://pytorch.org/get-started/previous-versions/

これでDreamboothのトレーニングの動作を確認出来た。
しかし暑いのでstability.AIの倫理的に問題の無いものであればColabでやった方が良いかも。

pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html