使用PyTorch撰寫CNN模型程式(MNIST資料)@Google Colaboratory平台(GPU)

動機:如果要使用PyTorch來撰寫CNN程式(MNIST資料) ,要如何實作呢?!


準備環境
1.Google Colaboratory 帳號
2.Google Chrome


實作步驟
1.首先,登入 Google Colaboratory,如下圖
 

2.其次,於 上方選單: [執行階段],選擇 //變更執行階段類型//,於 "硬體加速器" 拉選 GPU 再按下 [儲存]

3.再於 上方選單: [檔案],選擇 //上傳筆記本//,如下圖

4.最後,於 上方選單: [執行階段],選擇 //全部執行//,如下圖,花費時間約90秒(GPU)、360秒(CPU)


心得
1.使用了一年多的PyTorch,也經歷了其改版過程,發現當初沒有用TensorFlow是對的,此時此刻在向實驗室夥伴Demo(精簡版) Coding SOP之前,留下記錄...
2.本示範使用Google Colaboratory,可免安裝相關Python、Jupyter-notebook等執行環境
3.佛心的Google Colaboratory使用GPU是有限制時數的,若使用CPU執行也僅需約360秒即可...


參攷:本範例大多參考自網路分享的程式碼及內容,由衷感謝大家!!!


留言