search

google-research / l2p

Learning to Prompt (L2P) for Continual Learning @ CVPR22 and DualPrompt: Complementary Prompting for Rehearsal-free Continual Learning @ ECCV22
https://arxiv.org/pdf/2112.08654.pdf
Apache License 2.0
416 stars 42 forks source link

Only can run in CPU #2

Closed miss-rain closed 2 years ago

miss-rain commented 2 years ago

Both in Google Colab with TPU and my own Unbuntu with Cuda,

your code only run in CPU, not run in GPU or TPU!

zizhaozhang commented 2 years ago

@miss-rain I think it is related to your GPU type and cuda version and whether are them compatible with the installed jax version. It is not related to this code particularly.

Could you pls first make sure you can install jax and use jax to check GPU is detected (using this https://github.com/google/jax/issues/971).

KingSpencer commented 2 years ago

Please refer to issue #1.