Torch Device Cuda 0 If Cuda Else Cpu