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