通过谷歌云端硬盘(https://drive.google.com/drive/my-drive)使用Google Colab.
先将数据和代码上传到云端硬盘,然后右键选择Google Colab。进入后,可以将云端硬盘挂载上。因为默认使用的是CPU,可以在此处更改:代码执行情况->更改运行时的类型->选择GPU
运行代码前需要修改数据的位置,然后运行脚本:
!python3 drive/My\ Drive/SR-master/codes/demo_run.py
存储中间结果时,需要使用绝对路径,采用相对路径无法有效存在云端硬盘中,如下:
self.dir = '/content/drive/My Drive/Super-Resolution-Toolbox/experiment/' + args.save
查看分配的GPU型号和显存使用情况:
!/opt/bin/nvidia-smi
需要注意的是,分配到的GPU的型号是不一定的,而且如果最近使用很频繁,分配到GPU的概率也会变低,所以Google Colab只是适合于应急,很久没有使用时,临时来跑一下,如果长时间跑的话,很大可能后面只能使用cpu了。