Skip to content
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

Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR #486

Merged
merged 1 commit into from Nov 30, 2020

Conversation

cguegi
Copy link
Contributor

@cguegi cguegi commented Nov 25, 2020

When I'm running on TF 2.3.1 with the GPU, I am encountering
2020-11-25 09:06:12.391615: E tensorflow/stream_executor/cuda/cuda_dnn.cc:328] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

According to this issue the solution is to turn on memory growth.

This PR has been tested on Hadoop 3.1 with TF 2.3.1 and Cuda 10.2 on Nvidia Tesla T4

@oliverhu oliverhu merged commit 7672618 into tony-framework:master Nov 30, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants