使用nohup和CUDA_VISIBLE_DEVICES进行GPU训练的教程

在机器学习和深度学习中，训练模型时经常需要使用GPU来加速计算。本文将介绍如何使用nohup命令和CUDA_VISIBLE_DEVICES环境变量在后台运行Python脚本，并指定使用特定的GPU进行训练。

1. 在单个GPU上训练模型

如果你只有一个GPU，或者只需要在一个GPU上运行一个程序，可以按照以下步骤进行操作：

1.1 使用nohup命令运行Python脚本

nohup命令可以在用户退出会话后继续运行指定的命令。以下命令将Python脚本放到后台运行，并将输出重定向到nohup.out文件：

nohup python xxx.py &

1.2 查看运行中的进程

使用ps命令和grep可以查看当前运行的Python进程：

ps -ef | grep python