Ubuntu18.04LTS日志14-Pytorch编程常用功能

Ubuntu18.04LTS日志14-Pytorch编程常用功能

（1）选择指定GPU跑程序

// choose your GPU number
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'

（2）打印网络，查看网络层级结构

//查看网络结构
print（model）

//查看没一层的参数个数和输出个数
   params = list(model.parameters())
    k = 0
    for i in params:
        l = 1
        print("该层的结构：" + str(list(i.size())))
        for j in i.size():
            l *= j
        print("该层参数和：" + str(l))
        k = k + l
    print("总参数数量和：" + str(k))

（3）数据均衡到几个GPU 上进行训练

//假如有四个GPU
device_ids = [0,1,2,3]
model = nn.DataParallel(model, device_ids=device_ids)

（4）if name == ‘main’:的理解

//
if __name__ == '__main__':
	a = 1

这个语句代表当前程序如果为主程序执行的时候，a = 1执行。如果当前程序作为一个模块调用的时候，a = 1不执行。