再薅谷歌羊毛:用谷歌GPU免費訓練你的機器學習模型
上周谷歌推出的免費《機器學習速成課》讓很多人大呼美滋滋,因為這服務到家的中文版實在是太貼心了,重點是不花錢。(就是人工智慧合成的中文語音聽起來有點辣耳朵)
如何評價谷歌上線的基於 TensorFlow 的機器學習速成課程(免費中文版)?昨天有個叫 Nick Bourdakos 的程序員小哥在 medium 上分享了一個能讓大家再次薅谷歌羊毛的方法:免費用谷歌的GPU訓練你的機器學習模型。
嗯...對於這種薅羊毛的行為,必須轉發出來供大家批判一番!
眾所周知,在機器學習過程中,訓練模型通常是最耗時耗力的部分。但在 GPU 上訓練模型能讓訓練速度提高達 40 倍,將原本需要花費 2 天的訓練時間縮短至幾個小時。當然,想要用 GPU 實現這個效果,你得從兜里掏出點錢出來。
有天我(原作者Nick Bourdakos——譯者注)發現了一個很好用的工具叫 Google Colab,我把它描述為相當於 Jupyter notebook 的 Google 文件吧。Colab 的初衷是成為一個教育和研究工具,用於機器學習項目方面的合作。它有個令人喜大普奔的優點:完全永久免費!
使用這個工具不需要任何設置,我甚至都沒額外登錄(當然我已經登錄了我的 Google 賬戶)。
使用 Colab 能讓你免費獲取 12 個小時的時間無間斷使用一個 k80 GPU,功能很強大(雖然 12 個小時後就掉線了,但是你想用可以接著再用,想用多少回就用多少回)。
我想重點講講怎麼用 GPU 和 Colab 訓練模型,所以 notebook 這部分就一筆帶過了。
第一步是去下載 notebook
https://gist.github.com/bourdakos1/817611ebfe0d72a027ced9b072ec5c87當然你選擇別的 notebook 也行。然後直奔 Google Colab
https://colab.research.google.com/notebooks/welcome.ipynb#recent=true登上你的 Google 賬戶(要是你沒有 Google 賬戶,那就想辦法有一個)。
選擇 File > Upload notebook... :
上傳你下載的 notebook:
選擇 Runtime > Change runtime type:
然後選擇 GPU:
現在你應該能像正常一樣運行你的 notebook 了,唯一不同之處是最後一部分。如果你想通過瀏覽器下載你的模型或其它文件,可以用它們的 Python 程序庫:
from google.colab import filesfiles.download("PATH/TO/FILE")
後話
本文雖短,但我希望能幫助到大家,減少再在自己的小破筆記本上痛苦的等候模型訓練,也不必花一大筆銀子自己配置硬體。
羊毛雖好,但薅起來也注意節制喔。
參考資料:
https://hackernoon.com/train-your-machine-learning-models-on-googles-gpus-for-free-forever-a41bd309d6ad
推薦閱讀:
※3分鐘好像太狂了,那5分鐘幫你開啟機器學習的大門吧!
※《天才槍手》要失業?AI讓作弊更容易!
※[Python入門] 05 元組與資料庫
※邁向深度學習的第一步!零基礎深度學習:感知機
※聽說你想要可以抱著睡覺的人形機器人