Ignores my Nvidia Card?

[Saquall]

Hi, could someone give me a hand and help me getting this running on my Nvidia Card? I've followed the steps for it but when starting the process it tells me "Initializing KokoClone on: CPU"

So it seems to ignore my GPU? How would I check why it comes to this decision and is there a way to force GPU usage?

I've used https://rhulha.github.io/StreamingKokoroJS/ and it works my GPU so I know that this not a general problem.

Thanks!

[Ashish-Patnaik]

Hey @Saquall ! Thanks for trying out KokoClone. The "Initializing KokoClone on: CPU" message means that your Python environment currently cannot see CUDA, so torch.cuda.is_available() is returning False.

This usually happens if the standard CPU version of PyTorch was installed by mistake.

Could you try running this in your terminal (make sure your conda environment is activated)?
python -c "import torch; print(torch.cuda.is_available())"

If it says False, you need to install the CUDA version of PyTorch. You can fix this by running:
pip uninstall torch torchaudio
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu121

Let me know if that fixes it!