使用google的免費GPU運行機器學習代碼

原創地址

若是你想學習機器學習,可是又沒有一個本身的環境/顯卡.這裏告訴你們一個好消息,google上面有一個免費的機器學習環境能夠使用了.

在google搜索google colab,打開搜索到的地址:https://colab.research.google...

默認會提示建立一個新的記事本.這是由於google提供的是一個筆記本的環境,全部的代碼都是存儲在一個個的記事本中的.

這個時候能夠點擊取消查看默認的案例.

從案例中能夠看到代碼和文檔是直接結合在一塊兒的,同時能夠在代碼的最下面看到執行的結果.這裏稍微注意下一下,每段代碼都是單獨的一行.

能夠在以前的界面點擊建立,也能夠在左上角的文件中選擇建立.可選的有python2和python3這兩種版本,根據本身的喜愛來選擇就好.

建立以後記得選擇執行代碼程序下的更改運行時類型.

在下面的硬件加速器中選擇GPU,這樣以後的代碼就運行在gpu上了.不再怕家裏的電腦沒gpu了.

這裏添加一行代碼,輸入下面的一個案例.而後點擊左側的運行按鈕,這樣就能夠在下面看到運行的結果了.

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(20)
y = [x_i + np.random.randn(1) for x_i in x]
a, b = np.polyfit(x, y, 1)
plt.plot(x, y, 'o', np.arange(20), a*np.arange(20)+b, '-');