Tensorflow指定GPU的指定和显存设置

查看机器上GPU情况 命令: nvidia-smi 功能:显示机器上gpu的情况   命令: nvidia-smi -l 功能:定时更新显示机器上gpu的情况 其中左上侧有0、1、2、3的编号,表示GPU的编号,在后面指定GPU时需要使用这个编号。   在终端执行程序时指定GPU    CUDA_VISIBLE_DEVICES=1   python  your_file.py 这样在跑你的网络之前
相关文章
相关标签/搜索