PyTorch入门(四)使用GPU服务器炼丹

使用GPU服务器炼丹

      • Ubuntu后台运行、查看、 终止python文件
      • Ubuntu实时查看Nvidia显卡显存占用情况
      • 客户端使用socket连接服务器

Ubuntu后台运行、查看、 终止python文件

  • 后台运行
nohup python -u main.py >out.log 2>&1 &

nohup: 不挂断运行进程,就是当客户端断开ssh连接时,进程仍然可以继续在服务器运行
python -u main.py: command命令,运行main.py。其中,-u指令是把运行输出直接输出到stdout
>out.log 2>&1: 将python文件运行输出和错误输出重定向到out.log日志文件中
&: 后台运行进程
  • 查看
ps -aux
这个指令可以查看系统中正在运行的进程
  • 终止
kill -9 PID(进程号)

Ubuntu实时查看Nvidia显卡显存占用情况

终端命令

watch -n 10 nvidia-smi

其中,10表示每隔10秒刷新一次终端的显示结果。

客户端使用socket连接服务器

建立ssh隧道本地端口转发

ssh -N -f -L 20801:172.17.0.2:20801 [email protected]  -p 20801

-L: 本地端口转发,将本地的20801端口通过中间机器(202.120.38.4)
    转发到目标机器(172.17.0.2)的20801端口
-N :告诉SSH客户端,这个连接不需要执行任何命令,仅仅做端口转发
-f :后台运行SSH客户端

客户端socket : HOST = ‘127.0.0.1’  PORT = 20801
服务器socket : HOST = ' '          PORT = 20801

你可能感兴趣的:(PyTorch)