程序员

CV项目中常用Linux命令

2019-02-21  本文已影响0人  蛋黄也可以很有派

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

上一篇 下一篇

猜你喜欢

热点阅读