Quote from message that I received from Nick Marino:
If you just do pip install torch, that installs the CPU-only version of pytorch. If you use this:
pip3 install torch torchvision torchaudio --index-url [https://download.pytorch.org/whl/cu121](https://download.pytorch.org/whl/cu121%60)
That will download the CUDA accelerated version. You can see that here: https://pytorch.org/get-started/locally/
Quote from message that I received from Nick Marino:
If you just do pip install torch, that installs the CPU-only version of pytorch. If you use this:
pip3 install torch torchvision torchaudio --index-url [https://download.pytorch.org/whl/cu121](https://download.pytorch.org/whl/cu121%60)That will download the CUDA accelerated version. You can see that here: https://pytorch.org/get-started/locally/