多人使用多GPU系统指定GPU,防止冲突

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/csuzhaoqinghui/article/details/61928090
可以采取在当前用户下设置 vim ~/.bashrc
加入你想要使用的gpu的编号如下所示:
export CUDA_VISIBLE_DEVICES=1  Only device 1 will be seen
export CUDA_VISIBLE_DEVICES=0,1    Devices 0 and 1 will be visible
export CUDA_VISIBLE_DEVICES=”0,1  Same as above, quotation marks are optional
export CUDA_VISIBLE_DEVICES=0,2,3  Devices 0, 2, 3 will be visible; device 1 is masked
至此问题解决 主要针对 tensorflow   而theano可以在全局里面设置 即vim~/.theanorc

展开阅读全文

没有更多推荐了,返回首页