# 百度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.

打扰了。

最后更新于: 6/25/2020, 2:10:06 PM