-
Notifications
You must be signed in to change notification settings - Fork 344
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Error when running on Windows 10 #118
Comments
Do you use cuda 10.0 with cudnn? |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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)
The text was updated successfully, but these errors were encountered: