Tensorflow默认使用的是当前系统中的第一个可用GPU。如果系统中有多个GPU,可以通过设置环境变量或使用Tensorflow的API来指定使用特定的GPU。
在Tensorflow中,可以通过以下方式来指定使用的GPU:
import tensorflow as tf
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_visible_devices(physical_devices[0], 'GPU')
这样,Tensorflow将只使用第一个GPU进行计算。
需要注意的是,以上方法只是在Tensorflow内部限制了使用的GPU,并不会禁用其他GPU。如果系统中有其他进程正在使用GPU资源,仍然可能会导致Tensorflow无法使用指定的GPU。
推荐的腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云