如何使用特定显卡跑tensorflow

如何使用特定显卡跑tensorflow

直入正题，有两种比较简单的方法

方法一：改变系统变量使得仅目标显卡可见

export CUDA_VISIBLE_DEVICES=0 #这里是要使用的GPU编号，正常的话是从0开始

然后再运行tensorflow即可

方法二：使用tensorflow api添加设备环境

with tensorflow.device('/gpu:0'):
    main tensorflow program

如果想查看这种方法的官方手册，请戳这里

多说一句：

不管你通过方法一让哪一块或多块显卡可见，则在tensorflow的设备编号里这些可见卡的最小编号便变为“/gpu:0”。
此外，