Recently I’ve been studying machine learning. Since I’m an iOS developer I have tried Turi Create as well. I don’t own a GPU so to save time from training model so I used Google Colab. But I had quite hard time trying to run GPU with Turi Create. I have tried following guides on Google but none of them worked:
- turicreate-activity-classification.ipynb – Colaboratory
- turicreate-recommender.ipynb
- Training Turi Create models in Google Colab
- Using Google Colabs for Free TuriCreate Object Detection Training On The Cloud
- Turi Create Advanced Usage
After a few days I have figured out the way to make it works. So I write this post to share this experience with you.
I figured out that some versions of Turi Create work with GPU, some do not. Here is list of all Turi Create versions available that works with GPU on Google Colab and not works so far:
| Turi Create Version | Work with GPU |
|---|---|
| 6.4.1 | ❌ |
| 6.4 | ❌ |
| 6.3 | ❌ |
| 6.2.2 | ❌ |
| 6.2.1 | ❌ |
| 6.2 | ❌ |
| 6.1 | ✅ |
| 6.0 | ✅ |
| 5.7.1 | ❌ |
| 5.7 | ❌ |
| 4.2 | ❌ |
As the above table points out there are only 2 versions of Turi Create that work with GPU: 6.1 and 6.0. So all you have to do is specify version when installing Turi Create:
# Replace specific version you want to work with
!pip install turicreate==6.1If you do correctly, when you try to train model Google Colab’s log will show message “Using 1 GPU to create model” like the following picture:
And that’s all you have to do to run Turi Create with GPU on Google Colab.