使用Google-Colab訓練PyTorch神經網路
阿新 • • 發佈:2019-01-07
Colaboratory 是免費的 Jupyter 筆記本環境,不需要進行任何設定就可以使用,並且完全在雲端執行。關鍵是還有免費的GPU可以使用!用Colab訓練PyTorch神經網路步驟如下:
1:新建Colab檔案
Colab是在Google硬碟上面執行的,所以,需要到Google硬碟上面新建Colaboratory檔案,並進行關聯,檔案是以ipynb結尾的Jupyter筆記本。下面有一些Jupyter筆記本的使用技巧,可以幫你更好的使用這個環境:
直接執行python程式碼
import random
x = random.random()
print(x)
以 !開頭的命令可以執行作業系統的指令
! pwd
! ls
! cd .. # 切換了外面的工作目錄,但是不切換當前程式的工作目錄
以 %開頭的命令表示魔法指令
%matplotlib inline # 內嵌畫圖
%cd .. # 切換當前程式的工作目錄,作用和os.chdir(path) 相同
2:安裝pytorch
安裝方法一:
! pip3 install torch torchvision # 省事
安裝方法二:
! uname -a # 檢視系統 x86_64 GNU/Linux ! python --version # 檢視python版本 Python 3.6.7 ! nvcc --version # 檢視cuda版本 release 9.2, V9.2.148
在pytoch下載頁面找到合適的版本,進行安裝
! pip install https://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
! pip install torchvision
3:掛載Google硬碟
在執行程式碼的時候需要從Google硬碟讀取資料,需要將結果儲存到硬碟當中,這個時候需要進行掛載,期間需要輸入一個驗證碼。
from google.colab import drive drive.mount('/content/drive/') # 將google硬碟掛載在/comtent/drive/目錄上面
4:設定GPU
只需要在筆記本中的:修改->筆記本設定 當中就可以找到GPU選項。
5:編寫程式碼,進行訓練
直接編寫程式碼進行訓練,要注意當前的工作目錄,可以使用%cd path
進行切換
目前colab不穩定,連線經常斷掉,所以當訓練的次數過多、時間過長的時候注意儲存實驗結果。
如何執行python檔案?
當然你也可以不用Jupyter筆記本,使用以.py結尾的python檔案,這個時候只需要執行下面的程式碼。我沒有測試在這種情況下是否能夠時候用GPU。
! python test.py