在TensorFlow中,可以通過以下方式指定GPU運行:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 指定使用第一個GPU
import tensorflow as tf
gpu_options = tf.GPUOptions(visible_device_list="0") # 指定使用第一個GPU
config = tf.ConfigProto(gpu_options=gpu_options)
sess = tf.Session(config=config)
import tensorflow as tf
with tf.device('/device:GPU:0'):
# 在第一個GPU上運行的代碼
import tensorflow as tf
strategy = tf.distribute.MirroredStrategy()
with strategy.scope():
# 在多個GPU上運行的代碼
注意:在使用GPU時,需要安裝對應版本的CUDA和cuDNN,并且確保TensorFlow安裝時支持GPU。