最近時間發現,如果用本機跑圖片分類的話,實在是跑不動,跑得太慢了,服務器也不快。之前在搞了一通cuda、cudnn之後,還是覺得慢,而且cuda對MacOS太不友好了。
在師兄推薦下嘗試了一下google的colab。
進入了之後可以在設置裏選擇GPU或者TPU:
速度確實會快很多。但是我感覺colab有一個挺不方便的一點就是,本地文件需要上傳,上傳有兩種方式。
第一個是在代碼中輸入:
from google.colab import files
uploaded = files.upload()
這樣可以直接選擇文件上傳。
第二種方法是直接先將文件傳到google drive。個人感覺這種方法更快。
其他的方面使用起來與jupyter notebook沒差。
如果對速度和連接時間由更多的要求的話,可以購買colab pro版本,一個月9.9刀,價位也還挺合理的。速度是真的很香。