赞
踩
可以采取在当前用户下设置 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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。