Skip to content

Commit

Permalink
fix: to use gpu in colab with default tensorflow
Browse files Browse the repository at this point in the history
  • Loading branch information
@JinFoolish
JinFoolish committed Feb 21, 2024
1 parent fd3a8db commit 6c10c2d
Showing 1 changed file with 1 addition and 549 deletions.
Loading

1 comment on commit 6c10c2d

@JinFoolish
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I wanted to train my own model on colab a few days ago. At first, I directly used the pip install ddsp command, but when I started training, the output showed that some files could not be found, and colab's gpu was in an unused state. So I changed the pip command, and now I have successfully used this notebook to train several models on colab, and can use colab's GPU normally. The notebook runs smoothly now!

Please sign in to comment.