My pytorch model works well on cpu except it is time-comsuming. I would like to try the GPU version of pytorch. I added with cuda.device(1): and declared the tensors with cuda. There raised the AssertionError: "Torch not compiled with CUDA enabled."
What should I do to fix this installation error?