-
Notifications
You must be signed in to change notification settings - Fork 195
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
CPU support #2
Comments
If you remove all my explicit conv2DDNNLayer calls (aliased as C2D) and replace them with standard Conv2D layer calls it should work, but keep in mind that those DeconvLayers are adapted from Radford's DCGAN and make explicit DNN calls, so you would need some workaround for that as I don't think the TransposedConv2D layer will work identically (what with the 5x5 kernels in the DeconvLayers). I've had trouble with it in the past. |
I've added support for not explicitly making DNN calls, so you should, presumably, be able to try running this on CPU. At the very least, it shouldn't crash. |
Looks like it still fails:
|
@dkandalov Change the dnn=True line on line 53 to dnn=False, and it won't attempt to import cuDNN. |
@ajbrock It appears the correct line for this change is 18. |
Yes, following a commit after the above comments, the proper line for this is now 18. |
Might want to update the README |
Hello
Thanks for releasing source code. I've got problem while running script on laptop without cuda gpu:
How can I enforce CPU mode?
The text was updated successfully, but these errors were encountered: