百度AI Studio配置并运行pytorch项目的体验

总之,体验我给7分(满分10分)。

优点:

  • GPU挺快的,当然这一点没有比较过,直观感受上还挺快的

  • 算力卡(GPU可用时长)给得跟不要钱似的,完全就是那种屠龙宝刀来就送的感觉,而且我这算力卡怎么越用越多……

缺点:

  • notebook里只能使用百度家的飞桨框架(虽然在命令行可以用其他框架,但不稳定)
  • 动不动掉线,一掉线装好的依赖就全没了……在网上看到一堆持久化的方法,但好像并没有用,我没有往持久化的目录里写数据的权限,也没法sudo。要是有什么好办法,请各位指教~~,然后我把评分改高一点~~。

因为它的缺陷特别败好感,所以我只能给个7分。


刚才谈到,他这个平台不能在notebook里运行别的框架,所以pytorch只能在命令行用。还好他装了3.7的py,可以用pip(没有conda)安装pytorch。但是在装之前,先换成清华的镜像,这样能快一点:

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

然后看一下平台GPU的版本,免得装错了pytorch的版本:

nvcc -V
# nvcc: NVIDIA (R) Cuda compiler driver
# Copyright (c) 2005-2018 NVIDIA Corporation
# Built on Tue_Jun_12_23:07:04_CDT_2018
# Cuda compilation tools, release 9.2, V9.2.148

然后查看官方文档,因为这里是9.2的cuda,所以需要这么安装:

pip install torch==1.5.0+cu92 torchvision==0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

如果装错了,运行的时候就会报错。会出现这个情况:

torch.cuda.is_available() = False

感觉上相当于就是驱动不匹配,没法使用GPU,所以一定要装对。

等一会,安装完了之后,在命令行输个python,然后import torch试试,如果可以正常运行,那就正式完成了,可以开始使用。不过,只能在命令行用,即使是装好了,也不能在notebook里用:

Cannot run import torch because of system compatibility. AI Studio prepared an entire environment based on PaddlePaddle already. Please use PaddlePaddle to build your own model or application.

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