-
-
Notifications
You must be signed in to change notification settings - Fork 66
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
VGG example fails on GPU #42
Comments
Hmm, seems to be a different issue: FluxML/Flux.jl#287 (comment) |
So I added CUDNN, but am still getting a similar error:
|
if helpful, here are the Flux tests:
The 4 failures are already documented: FluxML/Flux.jl#267. And the Metalhead tests:
|
Closed
The underlying Flux issue has been resolved so I'm closing this. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
The code does run successfully on CPU, but as soon as I get CuArrays involved I get this issue. Same problem if I try
x = rand(Float32, 224, 224, 3, 1); vgg(x);
Edit: here are my manifest & project files: https://gist.github.com/tbenst/f370a5baae6e120c5869f591d5006794 for Julia 1.1
The text was updated successfully, but these errors were encountered: