check GPU support for pytorch