Closed lyriccoder closed 3 years ago
Hi @lyriccoder , Thank you for your interest in our work.
If I remember correctly, in TF there is no need to set up the GPU device, it is supposed to be detected automatically.
I don't know why does it say failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED
.
It looks like a software/drivers version mismatch.
Anyway, I recommend using code2seq instead.
Best, Uri
Thank you very much for help. The problem was in tensorflow, it couldn't find device since it was used it in a separate process. Tensorflow can't work with multiprocess (uwsgi)
Thank you for providing the source code to your paper.
Could you pelase help to solve the issue? When I predict a function name, the model is not responding here (it stucks at the line below):
I see that my gpu is visible in tensorflow.
I have the following config: tf=2.0.0 Cuda compilation tools, release 10.0, V10.0.130 cudaNN 7.5
The error is the following:
Also, I tried to multiply matrices with tensorflow, it worked correctly with GPU enabled. Any ideas?
Could you please tell me where you set up the GPU device? I cannot find it