How to install pytorch with cuda capability for AI acceleration with Nvidia Tesla etc.. GPUs
The default version is CPU only, which usually is not what we want if we have a GPU:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
Now cuda enabled pytorch apps like whisper etc... can take advantage of CUDA finally!
Check it is enabled
False means it is not working, or at least your cuda/nvidia driver setup is installed and True means good.
python3 -c "import torch; print(torch.cuda.is_available())"
True
Tags:
install, pytorch, cuda, capability, ai, acceleration, nvidia, tesla, etc, gpusthe, default, cpu, gpu, pip, torch, torchvision, torchaudio, index, url, https, download, org, whl, cu, enabled, apps, whisper, installed, python, quot, import, is_available,