I am using PyTorch on XPS 15 with GTX1050Ti and CUDA 10.1
I installed pytorch using the command:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
However, cuda.is_available() throws False. I dug into the issue and figured out that pytorch can indeed find the right cuda library cudart64_101 but I also get the followng assertion error:
AssertionError(“Torch not compiled with CUDA enabled”)
This seems inconsistent. Can someone help me get this to work?
Also see:

image1986×1934 319 KB