Hey, I resolved this. It was a conflict between an older preinstalled version which was 0.4.1. I removed it and reinstalled the nigthly version of pytorch and it works smoothly now. Thank you for your help.
Edit:
When running: python -c “import torch; print(torch.version)” I get the 1.0 version but when importing in jupyter notebook the 0.4.1 version appears again.
Edit2: Resolved, the new conda env was missing an ipython kernel so jupyter was loading PyTorch from another env that had one