latent-diffusion icon indicating copy to clipboard operation
latent-diffusion copied to clipboard
verified publisher icon CompVis

how to work on RTX 3090

Open adsf0427 opened this issue 3 years ago • 1 comments

When I follow the instruction, this occurs: NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.

adsf0427 avatar Jul 09 '22 03:07 adsf0427

Install the CUDA 11.3+ version of PyTorch --

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

See here - https://pytorch.org/

srelbo avatar Jul 09 '22 07:07 srelbo