深度学习常见代码汇总

1. 选择要用的GPU: CUDA_VISIBLE_DEVICES

服务器中有多个GPU，选择特定的GPU运行程序可在程序运行命令前使用：CUDA_VISIBLE_DEVICES=0命令。0为服务器中的GPU编号，可以为0, 1, 2, 3等，表明对程序可见的GPU编号。

2.os.environ

设置系统环境变量 ：os.environ[‘环境变量名称’]=‘环境变量值’

获取系统环境变量：os.environ[‘环境变量名称’] 

其中 key 和 value 均为 string 类型

查看所有环境变量的key：os.environ.keys()

3. seed

 random.seed(seed)
 np.random.seed(seed)
 torch.manual_seed(seed)
 torch.cuda.manual_seed_all(seed)

seed的具体作用详见

https://blog.youkuaiyun.com/lly_zy/article/details/93301908?ops_request_misc=&request_id=&biz_id=102&utm_term=seed&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-0-93301908