Updating Pytorch through Anaconda

I installed Pytorch in my anaconda environment using

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly

Using the nightly build I want to make sure I have the latest version after a couple of weeks so I did:

conda update --all

But then I get: pytorch pytorch-nightly::pytorch-1.6.0.dev202~ --> pkgs/main::pytorch-1.3.1-cpu_py37h62f834f_0 and the version when printing torch.__version__ is 1.3.1. What did I do wrong?

EDIT: When reinstalling pytorch nightly and the stable version on a new environment cuda wasn’t available, so that was very odd also. Before I got this weird behavior I had done sudo apt-get update, sudo apt-get upgrade but hadn’t rebooted the machine. After rebooting and reinstalling cuda was now available and everything seems to work as intended. Lesson learned: If they ask you to reboot then just reboot.

1 Like