CUDA_VISIBLE_DEVICES=0是指定GPU的编号， > 是将文件的输出重定向到文件 CUDA_VISIBLE_DEVICES=0 nohup python train.py > nohup1.txt 参考文献 [1][日常] Linux 后台指定 GPU 运行 PyTorch 代码
本文介绍了如何在Linux环境下，通过设置CUDA_VISIBLE_DEVICES环境变量为0来指定使用第一块GPU，并使用nohup命令将Python训练脚本（train.py）的输出重定向到nohup1.txt文件，实现后台运行并记录日志。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
