1. 程式人生 > >使用Google-Colab訓練PyTorch神經網路

使用Google-Colab訓練PyTorch神經網路

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