找不到 TF_MIN_GPU_MULTIPROCESSOR_COUNT

Can't find TF_MIN_GPU_MULTIPROCESSOR_COUNT

我收到一条消息,说我的 GPU 设备被忽略,因为它的多处理器数量低于最小设置。但是,它给了我环境变量 TF_MIN_GPU_MULTIPROCESSOR_COUNT 但它似乎不存在,因为我一直找不到命令。当我使用 setprintenv 和 grep 变量名称查看环境变量时,它不存在。有谁知道我在哪里可以找到它或如何更改它的设置值?

在运行你的主脚本之前做这样的事情 export TF_MIN_GPU_MULTIPROCESSOR_COUNT=4

请注意,默认设置是有原因的——如果您通过更改该变量启用较慢的 GPU,您的程序可能 运行 比没有可用 GPU 时慢,因为 TensorFlow 会尝试将运行 该 GPU 上的所有内容

在windows中,用这个名字创建一个新的环境变量并赋值。

您可以通过在 文件资源管理器 、select 中右键单击 This PC 来完成此操作底部的属性,然后左侧的select 高级系统设置。这将使您进入“系统属性”对话框。您也可以在 Cortana 搜索中键入 "environmental properties"。

从那里单击环境变量 按钮。一旦进入环境变量对话框,select新建变量并赋值,然后退出。您可能需要重新启动 IDE 或打开一个新的 DOS window 才能使该环境变量可见。