Stack trace after setting os.environ['CUDA_LAUNCH_BLOCKING'] = "1"
RuntimeError                              Traceback (most recent call last)
<ipython-input-6-6e6818118f97> in <module>()
     52             # run discriminator against randn draws
     53             z = torch.randn(images.size()[0], args['n_z']) * args['sigma']
---> 54             d_z = discriminator(z)
     55 
     56             # run discriminator against encoder z's

6 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/functional.py in linear(input, weight, bias)
   1672     if input.dim() == 2 and bias is not None:
   1673         # fused op is marginally faster
-> 1674         ret = torch.addmm(bias, input, weight.t())
   1675     else:
   1676         output = input.matmul(weight.t())

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)`