tensorflow新版本支持直接调用gpu,不需要专门安装tensorflow的GPU版本
安装以后,默认有GPU会直接调用GPU,
当显卡不够时,有其他不是很吃速度的任务想要训练,就需要切换CPU训练方式
os.environ["CUDA_VISIBLE_DEVICES"] = ""
多个显卡,指定特定GPU训练
os.environ['CUDA_VISIBLE_DEVICES']='2'
tensorflow新版本支持直接调用gpu,不需要专门安装tensorflow的GPU版本
安装以后,默认有GPU会直接调用GPU,
当显卡不够时,有其他不是很吃速度的任务想要训练,就需要切换CPU训练方式
os.environ["CUDA_VISIBLE_DEVICES"] = ""
多个显卡,指定特定GPU训练
os.environ['CUDA_VISIBLE_DEVICES']='2'