AIStudio怎么使用gpu训练
收藏
我使用的至尊版环境,用的是paddlex进行训练,但是gpu性能监控里接近0,cpu快满了,这个是自动使用gpu吗?还是需要代码?
需要代码的话,代码是什么?
我已经尝试了
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
还是没有使用gpu
0
收藏
请登录后评论
建议找几个paddlex的精品项目看一下 应该需要配个参数
paddlex在aistudio上不用配置吧?直接就是gpu了
可以nvidia-smi(在终端执行)看下GPU有没有跑起来
请问解决了吗?我也遇到这个问题了
你试试BML里运行
求问解决了嘛,安装cuda也安装不上去
应该是自带CUDA
不是import os,os.environ['CUDA_VISIBLE_DEVICES'] = '0',而是:export CUDA_VISIBLE_DEVICES=0
因为aistudio上是linux的环境。
建议重启项目,删除使用gpu的代码,它好像默认使用gpu,但是加了就会报错
如果gpu没跑起来该怎么办啊 是off的状态