-
Notifications
You must be signed in to change notification settings - Fork 16
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
Not running with GPU #25
Comments
Hi Yunjie, Does TensorFlow see the GPU on your cluster node where you are running the training? I would recommend to start an interactive cluster session and then check if the GPU is available with
Cheers! |
Hi Tim,
|
Hi Tim,
I just installed cryoCARE on our HPC following the installation procedure "For CUDA 10" and did not meet any errors during the installation.
However, I got the following message when I tried to run the training process (cryoCARE_train.py --conf train_config.json):
This information says that cryoCARE is not using GPU to do the training, instead it is using CPU, therefore, it is quite slow.
My tomogram size is 672672200.
Any idea about this issue?
Thanks!
Yunjie
The text was updated successfully, but these errors were encountered: