总结
- 给出模型训练中常用的命令
- 持续更新~
1 指定显卡训练
指定显卡训练有多种方法，这里简单列举两种。
1.1 在程序头添加
有时候，如果我们想在程序运行时就指定使用哪一块显卡，就可以在程序头部添加如下命令：
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “2”1.2 在运行的命令前添加
CUDA_VISIBLE_DEVICES=0 这就代表使用第0块显卡。
2 查看显卡状态
nvidia-smi
查看显卡当前使用状态
watch -n2 nvidia-smi ： 每隔2s 刷新 nvidia-smi 命令得到得输出
