python代码小技巧

1. 指定GPU运行

import os 
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"#指定0，1号GPU上运行代码

2. 查看GPU占用率

watch -n 1 -d nvidia-smi #每间隔1s刷新显示

3. 打印设计网络参数量

print ("Total number of paramerters in networks is {}  ".format (sum (x.numel () for x in model.parameters ())))