https://instantpost.co.uk/assertionerror-torch-not-compiled-with-cuda-enabled/