how do you fix "AssertionError: Torch not compiled with CUDA enabled" and " ModuleNotFoundError: No module named 'torch._six' "