美文网首页
【Linux备忘录】配置可使用的gpu资源

【Linux备忘录】配置可使用的gpu资源

作者: Ra1nyHouse | 来源:发表于2017-03-02 20:25 被阅读36次

通常实验室的gpu资源有限,而tensorflow默认使用gpu0,会造成冲突。如何设置只使用部分gpu资源呢,使用下列命令即可:

CUDA_VISIBLE_DEVICES=1 python yourfile.py

或者在.zshrc(.bash_profile)文件中添加

# 配置可使用的gpu, eg CUDA_VISIBLE_DEVICES=0,1
export CUDA_VISIBLE_DEVICES=1

相关文章

网友评论

      本文标题:【Linux备忘录】配置可使用的gpu资源

      本文链接:https://www.haomeiwen.com/subject/gzmkgttx.html