Assertionerror: Torch Not Compiled With Cuda Enabled With Code Examples