Error when running on Windows 10

[leeisack]

If you run python main.py , there is no problem at first. But when the ui comes out and clicks after that, it stops.
The error message is this:

(styleflow) PS C:\Users\isack\Documents\StyleFlow> python main.py
----------------- Options ---------------
batchSize: 1
checkpoints_dir: ./checkpoints
dataroot: ./data/datasetX
gpu_ids: 0
max_result_snapshots: 30
model: xxxx
name: XXXX
network_pkl: gdrive:networks/stylegan2-ffhq-config-f.pkl
only_for_test: ...
phase: test
----------------- End -------------------
Loading networks from "gdrive:networks/stylegan2-ffhq-config-f.pkl"...
Setting up TensorFlow plugin "fused_bias_act.cu": Preprocessing... Loading... Done.
Setting up TensorFlow plugin "upfirdn_2d.cu": Preprocessing... Loading... Done.
Training T : 1.0
Number of trainable parameters of Point CNF: 1691649
Traceback (most recent call last):
File "C:\Users\isack\Documents\StyleFlow\ui\mouse_event.py", line 39, in mousePressEvent
self.showImage(self.point)
File "C:\Users\isack\Documents\StyleFlow\ui\mouse_event.py", line 71, in showImage
self.Form.update_GT_scene_image()
File "main.py", line 158, in update_GT_scene_image
self.fws = self.prior(self.q_array, self.final_array_source, self.zero_padding)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "C:\Users\isack\Documents\StyleFlow\module\cnf.py", line 30, in forward
x, logpx = self.chain[i](x, context, logpx, integration_times, reverse)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "C:\Users\isack\Documents\StyleFlow\module\cnf.py", line 105, in forward
method=self.test_solver,
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torchdiffeq_impl\adjoint.py", line 129, in odeint_adjoint
ys = OdeintAdjointMethod.apply(*y0, func, t, flat_params, rtol, atol, method, options)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torchdiffeq_impl\adjoint.py", line 18, in forward
ans = odeint(func, y0, t, rtol=rtol, atol=atol, method=method, options=options)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torchdiffeq_impl\odeint.py", line 76, in odeint
solution = solver.integrate(t)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torchdiffeq_impl\solvers.py", line 29, in integrate
self.before_integrate(t)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torchdiffeq_impl\dopri5.py", line 78, in before_integrate
f0 = self.func(t[0].type_as(self.y0[0]), self.y0)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "C:\Users\isack\Documents\StyleFlow\module\odefunc.py", line 133, in forward
dy = self.diffeq(tc, y)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "C:\Users\isack\Documents\StyleFlow\module\odefunc.py", line 97, in forward
dx = layer(context, dx)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "C:\Users\isack\Documents\StyleFlow\module\diffeq_layers.py", line 86, in forward
ret = self._layer(x) * gate + bias
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torch\nn\modules\module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torch\nn\modules\linear.py", line 87, in forward
return F.linear(input, self.weight, self.bias)
File "C:\Users\isack\Anaconda3\envs\styleflow\lib\site-packages\torch\nn\functional.py", line 1371, in linear
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)

[leeisack]

Additionally, my output in web ui.

[vicdxxx]

Do you use cuda 10.0 with cudnn?