TensorFlow使用技巧

来源:互联网 发布:ps4网络设置 编辑:程序博客网 时间:2024/05/21 21:37

1、限定TensorFlow使用的GPU

    即使目前的任务并不需要很多的GPU资源,TensorFlow在默认情况下任然会占用机器上所有的GPU资源,因此在使用TensorFlow之前,有必要对其使用的GPU资源进行限定。

    例如,使用python执行TensorFlow程序,在启动python之前加入如下语句,既可以限定使用GPU编号。

CUDA_VISIBLE_DEVICES=1

CUDA_VISIBLE_DEVICES=”0,1”

CUDA_VISIBLE_DEVICES=0,2,3





0 0
原创粉丝点击