颜娘娘的碎碎念

关注

tensorflow训练方式切换

颜娘娘的碎碎念

关注

阅读 29

2022-01-27

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

os.environ["CUDA_VISIBLE_DEVICES"] = ""

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

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

相关推荐

素锦时年_1b00

Tensorflow整理[5].Tensorflow运作方式

素锦时年_1b00 76 0 0

小月亮06

PPT切换方式

小月亮06 81 0 0

夏天的枫_

TensorFlow模型训练常见案例

夏天的枫_ 25 0 0

49路末班车

Tensorflow训练视觉模型(CPU)

49路末班车 6 0 0

大明宫

【Tensorflow】通过TensorFlow2.0训练神经网络模型

大明宫 112 0 0

泠之屋

TensorFlow深度学习并发加速训练

泠之屋 86 0 0

大自然在召唤

Tensorflow2.0笔记 - 不使用layer方式,简单的MNIST训练

大自然在召唤 12 0 0

千妈小语

OpenCV调用TensorFlow预训练模型

千妈小语 46 0 0

快乐与微笑的淘气

【Keras】TensorFlow分布式训练

快乐与微笑的淘气 66 0 0

豆丁趣

adaboost方式进行训练

豆丁趣 55 0 0

精彩评论(0)

0 0 举报