fine_tune albert with run_classifier.py on GPU but the GPU-util is 0,why and how?

google-research / albert

ALBERT: A Lite BERT for Self-supervised Learning of Language Representations

Apache License 2.0

3.23k stars 570 forks source link

fine_tune albert with run_classifier.py on GPU but the GPU-util is 0,why and how? #211

Open yuanjun1991 opened 4 years ago

ares5221 commented 4 years ago

your tensorflow is upgrade tf2.x?

JanKaul commented 2 years ago

Are you installing a different tensorflow version with pip in google colab? Then the GPU might not be set up properly. Instead use the %tensorflow_version magic as described here.