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.