Open1
GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
PyTorchのバージョンがCUDAのバージョンと合って無いときに下記のエラーが出るようだ。
GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
▼対処方法は下記のブログから知ることが出来た
▼pipもしくはcondaの環境ではcondaでCUDAバージョンに合致するPyTorchをインストールする。
これで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