会员
周边
新闻
博问
AI培训
云市场
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
理想几岁
博客园
新随笔
联系
管理
2020年2月18日
指定GPU训练模型
摘要: 公司或者实验室当大家都共用一台服务器时,训练模型的时候如果不指定GPU,往往会冲突。 我们可以先用 nvidia-smi 查看有多少块GPU, 然后分两种方式指定GPU运行。 1、直接在终端运行时加入相关语句实现指定GPU的使用 CUDA_VISIBLE_DEVICES=0 python test.
阅读全文
posted @ 2020-02-18 17:18 理想几岁
阅读(5394)
评论(0)
推荐(0)
编辑