爲什麼colab pro的GPU跑的很慢-解決方法

最近在colab上跑了一下cifar-10的圖像分類數據,結果發現跑的很慢。拿本機的CPU試了一下,一個epoch大概需要20min;在colab的GPU上甚至需要兩倍以上的時間。感覺很不合常理。

先確認了一下是否在使用GPU:


OK,確實是有在用cuda。百思不得其解之下,在Stack Overflow上發現了一個帖子:
https://stackoverflow.com/questions/60798910/google-colab-pro-gpu-running-extremely-slow

跟我的問題很像,帖子中有個回覆說drive是個雲盤(我的數據掛在到了google drive上),每次讀取數據都比較慢,可以拷貝到local路徑或者直接下載到local中。


我直接下載到了當前路徑,解壓:

!mkdir train_local
!wget https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
!tar zxvf cifar-10-python.tar.gz -C train_local
!rm cifar-10-python.tar.gz

重新訓練,速度快了非常多(原來一個epoch大概需要6min,現在只需要20sec。

但是這裏有個疑惑是,都用雲盤讀取的時候,用colab的GPU仍然比CPU(類型“None”)要慢?

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章