当前位置: 代码迷 >> 综合 >> GPU的使用
  详细解决方案

GPU的使用

热度:5   发布时间:2024-01-26 20:44:33.0

1、bash中

## bash
设定GPU卡顺序:export CUDA_DEVICE_ORDER = "PCI_BUS_ID"
指定第1块和第2块GPU卡:export CUDA_VISIBLE_DEVICES="0,1" 

2、python代码中

## pythonimport osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3"  ,这样就指定了第1,3,4,张卡可见了设置定量的GPU使用量
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction=0.9 # 占用GPU90%的显存
session = tf.Session(config=config)

 

  相关解决方案