在linux下用终端训练和测试模型

步骤

(1)在python中,指定要使用的gpu,代码如下:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"  　# 指定第一块gpu

(2)在py文件所在目录,打开终端运行py文件,命令如下:

CUDA_VISIBLE_DEVICES=0 python  main.py       # 终端运行py时,指定gpu 

(3)其它命令,如下:

CUDA_VISIBLE_DEVICES=0 nohup python main.py &     # 终端指定gpu,后台挂起
tailf nohup.out       # 查看nohup里的日志
ps x          # 查看进程

参考

https://blog.csdn.net/weixin_40277254/article/details/89387496