一、指定GPU运行程序

        因为条件限制，拿到的资源只有一块GPU，这时我们需要设置程序在指定的GPU运行。解决思路：在train文件中设置环境变量，让程序在指定GPU运行。

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "7" # run code on sepecific gpu

        测试一下，成了。

二、设置cmd运行的程序后台运行

        之前没有用过unbuntu服务器跑程序，以为在cmd窗口把程序运行起来，它会一直running。后来，当我兴致满满的再次上线的时候发现，他喵的，只跑了一个epoch？？？

        网上给的解答是：cmd连接的窗口一旦关闭，对应的程序（因为没有被配置为在后台运行或没有被守护进程管理）同时会被系统干掉。

        解决方法：使用tmux创建会话。tmux是终端复用器，允许你启动一个会话，即使远程连接断开，会话中的程序也会继续运行。重新连接时，你可以回到之前的会话。

tmux new -s session_name 
python your_script.py

        退出tmux会话时使用Ctrl+B然后按D，重新连接会话使用：

tmux attach -t session_name