Upgrading from CPU only to GPU

I have torch and torchvision installed through both conda and pip, although both are CPU based. Since I have a GPU now, I want conda version to be upgraded to GPU version. Any help on the steps to do it?

You can install the binaries with CUDA by selecting your OS, CUDA version etc. from the website.

1 Like

thanks for the quick response. doing that will remove cpu version as well?

I would guess so (haven’t tested it in the past weeks though).
If you want to keep multiple PyTorch versions, I would recommend to create virtual environments.