【环境配置】服务器后台指定 GPU 运行代码

  CUDA_VISIBLE_DEVICES=0是指定GPU的编号, > 是将文件的输出重定向到文件

CUDA_VISIBLE_DEVICES=0 nohup python train.py > nohup1.txt

参考文献

[1][日常] Linux 后台指定 GPU 运行 PyTorch 代码

Guess you like

Origin blog.csdn.net/zfhsfdhdfajhsr/article/details/120869036