Upgrading from CPU only to GPU

1

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?

2

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

1 Like
3

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

4

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.