Skip to content

A
AutoDL

PaddlePaddle / AutoDL

通知 3
Star 2
Fork 0
A
AutoDL
提交 d3883761 编写于 作者: Q qinduohao
浏览文件
操作

readme add CUDA_VISIBLE_DEVICES description

上级 7d2fd1ef
隐藏空白更改
内联 并排
Showing with 2 addition and 0 deletion
AutoDL Design/README.md
浏览文件 @ d3883761
...@@ -60,6 +60,8 @@ mid=460, average rewards=17.000 ...@@ -60,6 +60,8 @@ mid=460, average rewards=17.000
export FLAGS_fast_eager_deletion_mode=1 export FLAGS_fast_eager_deletion_mode=1
CUDA_VISIBLE_DEVICES=0 python -u main.py CUDA_VISIBLE_DEVICES=0 python -u main.py
``` ```
__注意:__ 这里训练需要使用两张卡来训练，Agent使用的卡为`CUDA_VISIBLE_DEVICES=0`(设置在启动`main.py`命令中)；Trainer训练使用卡为`CUDA_VISIBLE_DEVICES=1`(设置在[autodl.py](https://github.com/PaddlePaddle/AutoDL/blob/master/AutoDL%20Design/autodl.py#L124)文件中)
预期结果： 预期结果：
日志中 `average accuracy`逐步增大 日志中 `average accuracy`逐步增大
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或