使用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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM