Use GPU environment in Colab?

[Bob-YsPan]

I tried use this library in the Colab Notebook, and I choose a GPU environment to acclerate the process speed. But seems like it not detect any gpu that Colab provides, but I check the lspci command and tensorflow library it can detect a GPU:

[Bob-YsPan]

I found the solution We need to replace the driver that Colab provides to full-support driver, the problem fixes

(The command shows below need to add ! at command start or %%bash at first line to run correctly)

apt purge *nvidia*

then

apt install -y --no-install-recommends nvidia-driver-530