GPUmemory简介，FFT算法原理与实现

今天小编就为大家分享一篇tensorflow指定GPU与动态分配GPU memory设置，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

在tensorflow中，默认指定占用所有的GPU，如需指定占用的GPU，可以在命令行中： export CUDA_VISIBLE_DEVICES=1 这样便是只占用1号GPU，通过命令 nvidia-smi 可以查看各个GPU的使用情况。 另外，也可以在python程序中指定GPU，并且动态分配memory，代码如下 import os import sys os.environ['CUDA_VISIBLE_DEVICES'] = sys.argv[1] import tensorflo