在默认条件下,tensorflow会一次占满左右显存!这对于我们想在同一台机器上跑多个程序不利(后面的程序会报显存不足的错误),可以通过以下方式,设置tensorflow中显存的使用方式
在Python代码中指定GPU
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
os.environ["CUDA_VISIBLE_DEVICES"
在默认条件下,tensorflow会一次占满左右显存!这对于我们想在同一台机器上跑多个程序不利(后面的程序会报显存不足的错误),可以通过以下方式,设置tensorflow中显存的使用方式
在Python代码中指定GPU
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
os.environ["CUDA_VISIBLE_DEVICES"