0
点赞
收藏
分享

微信扫一扫

tensorflow训练方式切换

tensorflow新版本支持直接调用gpu,不需要专门安装tensorflow的GPU版本
安装以后,默认有GPU会直接调用GPU,
当显卡不够时,有其他不是很吃速度的任务想要训练,就需要切换CPU训练方式

os.environ["CUDA_VISIBLE_DEVICES"] = ""

多个显卡,指定特定GPU训练

os.environ['CUDA_VISIBLE_DEVICES']='2'
举报

相关推荐

0 条评论