search

kcosta42 / VQGAN-CLIP-Docker

Zero-Shot Text-to-Image Generation VQGAN+CLIP Dockerized
MIT License
74 stars 17 forks source link

GPU #6

Open IHateMax opened 2 years ago

IHateMax commented 2 years ago

Try as I might I keep getting FALSE for

python3 -c "import torch; print(torch.cuda.is_available());"

I have re-installed CUDA and all my drivers, I have a compatible card. Can't seem to figure out what is wrong.

python -c "import torch; print(torch.cuda.is_available());"

results in TRUE

I am fairly new to all of this so sorry if its a simple solution.

Cheers.