fatfatmomo的博客

原创 tensorflow 调用多块GPU的一些错误

服务器两块GPU，跑tensorflow 版bert想将两块GPU都用上。看了别的教程说有两种方法，在windows server 2016上试验了一下，用的GPU-Z工具进行GPU观察，因为windows server的任务管理器是个阉割版看不到GPU性能。 方法一： import os os.environ["CUDA_VISIBLE_DEVICES"] = "1"#指定显卡1运行 #...