tensorflow新版本支持直接调用gpu,不需要专门安装tensorflow的GPU版本
安装以后,默认有GPU会直接调用GPU,
当显卡不够时,有其他不是很吃速度的任务想要训练,就需要切换CPU训练方式
os.environ["CUDA_VISIBLE_DEVICES"] = ""
多个显卡,指定特定GPU训练
os.environ['CUDA_VISIBLE_DEVICES']='2'
tensorflow训练方式切换
阅读 29
2022-01-27
tensorflow新版本支持直接调用gpu,不需要专门安装tensorflow的GPU版本
安装以后,默认有GPU会直接调用GPU,
当显卡不够时,有其他不是很吃速度的任务想要训练,就需要切换CPU训练方式
os.environ["CUDA_VISIBLE_DEVICES"] = ""
多个显卡,指定特定GPU训练
os.environ['CUDA_VISIBLE_DEVICES']='2'
相关推荐
精彩评论(0)