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

2020.11.19 现在可能已经不能用了。有好几个人私信跟我说这个事,现在 torch 好像会闪退。

总之,体验我给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.

  • 5
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值