diffusers-rs icon indicating copy to clipboard operation
diffusers-rs copied to clipboard

Published 20 hours ago • verified publisher icon LaurentMazare

CUDA/GPU Not Working.

Open Johnmakuta opened this issue 2 years ago • 1 comments

I've been testing out the default test script and every time is says:

Cuda available: false
Cudnn available: false
MPS available: false

Even though I've testing in pytorch and tensorflow that my gpu is available and shown.

Johnmakuta avatar Jul 08 '23 05:07 Johnmakuta

This is very dependent in the way you've been installing the libtorch library. If you have a working install of PyTorch 2.0.0 on which cuda/cudnn work, I would recommend using it by activating the environment and setting LIBTORCH_USE_PYTORCH=1 (more details on how libtorch can be installed on the main tch readme.

LaurentMazare avatar Jul 08 '23 06:07 LaurentMazare