search

LaurentMazare / diffusers-rs

An implementation of the diffusers api in Rust
Apache License 2.0
535 stars 55 forks source link

CUDA/GPU Not Working. #74

Open Johnmakuta opened 1 year ago

Johnmakuta commented 1 year ago

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.

LaurentMazare commented 1 year ago

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.