1. GPU设置
使用GPU版本的TensorFlow跑程序默认是占用所有主机GPU,但计算过程中只会用其中一块。也就是说所有显卡都闲着,但其显存都被占用了,所以别人也用不了。
这种情况通过在程序之前加三行代码:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ['CUDA_VISIBLE_DEVICES'] = "0,1"
如不想使用GPU,第三行代码 = ""
nvidia-smi 观察一下GPU显存使用情况。
2. 杀死进程
kill -9 + pid
网友评论