-
Notifications
You must be signed in to change notification settings - Fork 709
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
GPU available but not used #232
Comments
Hi @alevangel, Which GPU model do you have on Colab? |
GPU 0: Tesla T4 (edited) |
@alevangel Can you share your torch version? |
|
I solved rolling back at version of pytorch-lightning==1.5.9 |
I have the same problem with CFLOW model on my local machine.
GPU:
|
@andriy-onufriyenko use pytorch-lightning==1.5.9 |
@alevangel Install pytorch-lightning==1.5.9 and have another problem now:
|
@andriy-onufriyenko, can you add |
It's working but without GPU
Trying edit
and got
Parameter |
I found the problem. PyTorch Lightning deprecated and / or modified some of the The issue has been fixed in PR #234 |
I checked it works for me:
Just replace all 11 files from PR 234 in to my project. |
Executing CFLOW model on Colab, PyTorch returns this
Even if i'm setting gpus=1 on the config.yml of cflow.
To Reproduce
train.py --model cflow
GPU Model
GPU 0: Tesla T4
Versions
The text was updated successfully, but these errors were encountered: