As the title says I am trying to get a newer version of pytorch to get this option in the torch.unique function. However when I do
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
I do not get the latest version of pytorch (just the 1.0), which apparently does not include this option.
How can I get the correct version ?
EDIT :
I of course tried :
conda install pytorch=1.2 torchvision cudatoolkit=10.1 -c pytorch
but I cannot compile C++ extension as it says the cuda installation is not found and there is no cuda_make_macros.h