I am new to Pytorch.  I have been running the sample code for reinforcment learning (https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html 1) on a different data set that has 5 output classes.  It runs fine on my Python 3.7 environment, but gives a runtime error when running on Python 3.6
THCudaCheck FAIL file=C:/cb/pytorch_1000000000000/work/aten/src\THC/generic/THCTensorMath.cu line=29 error=710 : device-side assert triggered
C:/cb/pytorch_1000000000000/work/aten/src/THCUNN/ClassNLLCriterion.cu:108: block: [0,0,0], thread: [0,0,0] Assertion t >= 0 && t < n_classes failed.
I’ve been comparing my environment for days and can’t see any tangible differences.  Any pointers in the right directions would be appreciated.