当我们电脑配置不够好,而代码工程又很大的时候,可以借助colab提供的这个平台来跑GPU程序,会提高代码运行的速度。
①打开 drive
②点击“上传文件夹”,把自己的工程文件夹上传,就可以在colab中使用了
③打开 colab ,点击代码执行程序--更改运行时类型,切换到GPU
新建笔记本,新建代码单元格
挂载 drive
from google.colab import drive
drive.mount('/content/gdrive')
点击小箭头运行,出现下面情况
进入链接,登录账号,粘贴代码到方框中,按下回车后,出现下图即为挂载成功
接着,新建一个代码单元格运行下面代码,注意替换为自己的工程名
import os
os.chdir("/content/gdrive/MyDrive/自己的工程名")
注意,如果不运行上面这个代码,我门就需要在上传的工程文件中,所有路径都使用绝对路径,写为
然后,我们就可以运行自己的主函数了
!python main.py
如果报错缺什么包,则在运行程序之前添加代码块安装对应的包,例如,缺少tensorflow,就需要添加:
!pip install --upgrade tensorflow
OK!