一般租的服务器会有多张显卡，为了资源最大化利用同时运行程序，如果不指定显卡运行会导致多个程序抢占一张显卡运行，然后报错内存占满。

我们先查看系统有多少张显卡，每张显卡的运行状态：

nvidia-smi

在终端输入上面的指令得到下面的结果，显示了系统安装的所有显卡和每张卡当前的状态，比如显卡0当前使用率为81%，已经比较接近占满了。
在这里插入图片描述
我们观察到1，2，3，4，5都没程序使用，于是指定其中几张，指令如下

CUDA_VISIBLE_DEVICES=3,4,1 python train.py

意思为同时用显卡3，4，1运行代码train.py，然后3为主显卡，4和1为辅助显卡。

然后运行即可。
但是我们知道，我们远程连接服务器，输入上面的指令程序运行过程会显示在终端，如果中途网络问题和服务器连接失败，终端可能会关闭，程序运行中断，于是我们想到如下指令：

nohup python train.py &

但是当我们写成

nohup CUDA_VISIBLE_DEVICES=3,4,1 python train.py &

会报错指令不对，所以这个时候最好的办法就是，在train.py程序中加上说明，在代码里面最开始的部分加上如下指令：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '3,4,1'

这里在代码里面调用终端指令，说明运行的显卡。

希望对你的学习有所帮助。

# 网络 # linux # 操作系统
Logo
Linux

更多推荐

网卡速率和双工模式的配置

http://linux.chinaitlab.com/system/792187.html1、mii-tool 配置网络设备协商方式的工具；    1.1 mii-tool 介绍；    mii-tool - view, manipulate media-independent interface status （mii-tool 是查看，管理介质的网络接口的状态）

avatar Linux

Linux虚拟文件系统之文件系统卸载（sys_umount())

Linux中卸载文件系统由umount系统调用实现，入口函数为sys_umount()。较于文件系统的安装较为简单，下面是具体的实现。1. /*sys_umont系统调用*/2. SYSCALL_DEFINE2(umount, char __user *, name, int, flags)3. {4.struct path path;

avatar Linux

Linux系统下超级终端Minicom的使用方法（例如：连接交换机，路由器等）转http://baike.baidu.com/view/2911642.htm?fr=ala0_1

<br />　　Linux系统下超级终端Minicom的使用方法 <br />　　Linux下的Minicom的功能与下的超级终端功能相似，适于在通过超级终端对设备的管理以及对嵌入操作系统的升级，现写出Minicom的使用手册： <br />　　1． 启动minicom <br />　　以root权限登录系统 <br />　　使用命令 <br />　　minicom –s 则minicom启动，屏

avatar Linux