多显卡服务器指定显卡运行tensorflow程序

查看机器上GPU情况 命令: nvidia-smi 功能:显示机器上gpu的情况 其中左上侧有0、1、2、3的编号,表示GPU的编号,在后面指定GPU时需要使用这个编号。 方法一:在终端执行程序时指定GPU   CUDA_VISIBLE_DEVICES=1   python  your_file.py 这样在跑你的网络之前,告诉程序只能看到1号GPU,其他的GPU它不可见 可用的形式如下: CUD
相关文章
相关标签/搜索