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

可以采取在当前用户下设置 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

你可能感兴趣的:(深度学习)