在ai studio中使用PaddleSeg训练时,GPU没有使用
收藏
用了高级模式,但训练时出现图中情况
训练时代码:
这是因为CPU能对付,然后GPU就没有使用了吗?求解答。
0
收藏
请登录后评论
应该不是的。
1. 首先得确保安装的是GPU版的PaddlePaddle;
2. 确保在配置文件中开启了使用GPU;
3. AI Studio上是linux环境,使用GPU之前必须使用命令:export CUDA_VISIBLE_DEVICES=0
是的