Torch enable cuda

Torch enable cudaTorch enable cudaTorch enable cudaTorch enable cudaTorch enable cudaTorch enable cudaTorch enable cudaTorch enable cudaTorch enable cudaTorch enable cuda