分类：GPU设置

2021
07-04

TensorFlow和keras中GPU使用的设置操作

1.训练运行时候指定GPU运行时候加一行代码：CUDA_VISIBLE_DEVICES=1pythontrain.py2.运行过程中按需或者定量分配GPUtensorflow直接在开启Session时候加几行代码就行，而Keras指定GPU，并限制按需用量和TensorFlow不太一样，因为keras训练是封装好的，不好对Session操作。如下是两种对应的操作。keras中的操作：importosimporttensorflowastffromkeras.backend.tensorflow_backendimportset_session#指定第一块GPU可用... 继续阅读 >

2021
03-05

神经网络训练采用gpu设置的方式

在定义图结构之前不用加入gpu：0，只有在session中计算之前在加入，否则的话会提示不能使用gpu保存模型等问题。withtf.device('/gpu:0'):补充：关于应用gpu训练神经网络的注意事项对于GPU来说，一定要注意的是，要分别在两个GPU上，或者不同时的在一个GPU上运行train和evaluation的部分，否则限于GPU擅长迭代而不擅长逻辑的特性，会发生OOM（outofmemory）以上为个人经验，希望能给大家一个参考，也希望大家多多支持自学编... 继续阅读 >