Skip to content

Conversation

@scatena
Copy link

@scatena scatena commented Aug 18, 2021

Training could not be reproduced out of the box under a Windows system. Python environment would point a problem initializing cuBLAS (and all further operations requiring it).
Borrowed a fix described at https://stackoverflow.com/a/52132342 and training was possible as expected.

Do note: I'm not not a TF expert myself, so I can't know if said inclusion would affect overall compatibility.
Similar issues all relate to memory growth. More recent documentation point to alternative solutions as in https://stackoverflow.com/a/60699372

Thanks for the generosity with the code!

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.

1 participant