【TensorFlow】TensorFlow指定特定GPU以及占用显存的比例
因为LZ是使用GPU服务器跑TensorFlow,而TensorFlow默认的是占用所有GPU,于是为了不影响其他同学使用GPU,于是就试验和总结了一下TensorFlow指定GPU的方法。。
环境
系统:Ubuntu14.04
TensorFlow:v1.3
GPU 8个GTX1080,第一列的0~7的数是GPU的序号
一.设置指定GPU
1.一劳永逸的方法,直接在~/.bashrc...
转载
2019-09-19 15:30:19 ·
444 阅读 ·
0 评论