Is there a simple function which tests the GPU is configured correctly?
As I did what @hughperkins suggested on the following MNIST example:

  
      github.com
  
  
    rickiepark/pytorch-examples/blob/master/mnist.ipynb 233
{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### PyTorch MNIST Example similar to TensorFlow Tutorial"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Refer to https://www.tensorflow.org/versions/master/tutorials/mnist/pros/"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [


  This file has been truncated. show original

  
  
    
    
  
  


Yet my system won’t run (Calculation will stop with error).
Thank You.