在GPU工作站/服务器上运行代码及常用操作

  • Post author:
  • Post category:其他


服务器连接

本人所在的实验室服务器是安装了Ubuntu系统的台式机

首先,服务器是支持ssh远程连接的。在你的PC上,打开命令行,即可输入指令连接。(windows平台可以使用Xshell)

输入命令:

ssh UserName@Server_IP

输入密码完成后回车进入服务器:

上传文件

推荐大家,打包上传代码(方便存档)

scp PATH_TO_LOCAL UserName@ServerIP:PATH_TO_SERVER

运行代码

使用unzip命令来解压代码包

cd 目录:

python XXX.py 运行代码

公用的服务器,如果想制定某一个GPU来运行,可以设置CUDA_VISIBLE_DEVICES。在调用python程序时,前面加一个参数。

CUDA_VISIBLE_DEVICES=1 python my_script.py

这就是使用第二个GPU来训练。



版权声明:本文为qq_34395746原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。