tensorflow/keras指定运行时显卡及限制GPU用量

tensorflow和基于tensorflow的keras运行时会自动占用所有gpu,为了限制程序的gpu占用量,可以指定程序只使用某一个gpu,参考:

https://blog.csdn.net/A632189007/article/details/77978058

其中可以在程序开头位置加入语句

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

来指定只使用第一块gpu

可以用语句

watch -n 1 nvidia-smi # 1秒钟刷新一次

来查看gpu使用情况。

之前我加入以上语句一直没有生效,最后发现原因是使用pycharm远程开发连接服务器的时候,虽然点了autoupload,但却没有自动上传,因此改了代码也没有同步 。所以还是取消自动更新,以后改完代码手动upload更稳妥。

改正之后,上面的语句是可以正确执行的,可以看到我的进程只占用了GPU0,而且是占用了GPU0的所有剩余容量

猜你喜欢

转载自blog.csdn.net/weixin_41519463/article/details/89960745