tensorflow使用gpu

显卡1060,cpu i7 7700,16G内存。
CUDA_VISIBLE_DEVICES=1 只使用gpu1
CUDA_VISIBLE_DEVICES=0,1 只用gpu0和gpu1
CUDA_VISIBLE_DEVICES=”0,1” 加引号与不加引号作用相同
CUDA_VISIBLE_DEVICES=0,2,3 只用gpu0,gpu2,gpu3不适用gpu1
CUDA_VISIBLE_DEVICES=“” 使用cpu

开始测试
CUDA_VISIBLE_DEVICES=0 python xxxx.py

测试结果,CNN使用gpu速度提高大约5倍。但是lstm没有提高,在工作模式时CPU比GPU快3倍。

部分内容出处http://www.acceleware.com/blog/cudavisibledevices-masking-gpus

发表评论

电子邮件地址不会被公开。 必填项已用*标注