Question

当我启动tensorflow Session时，它立即在我的dekstop（设备0和2）中的两个GPU上分配100％的内存。

如何使用Python API将其限制为仅1个GPU？

Answer 1

例如，限制它仅使用GPU＃0的粗略方法是定义：

export CUDA_VISIBLE_DEVICES=0

Answer 2

如果您希望仅使用特定设备执行某些操作，则可以执行以下操作：

foreach( $validSearchWords as $word ) { //this is a cleaned array of the user query phrase
    $mycount++;

    $word = str_replace('\'', '\'\'', $word);
    // if only one search term
    if ($vcount == $mycount) {
        $wordCriteria['keyword'] .= $word.'*';
        // if term is second or higher
    } else {
        $wordCriteria['keyword'] .= $word.'* ';
    }
}

以下列方式指定设备：with tf.device('/gpu:0'): ... # all your operations，"/cpu:0"，"/gpu:0"。我建议您阅读using GPU devices主题中的更多详细信息。

如果您想查看一些教程，请查看cifar10 tutorial.

如何在Python界面中指定GPU的数量？

2 个答案: