摘要：
1.Docker启动：dockerrun--rm--gpus0-p5005:5005Bert:latestdockerrun--rm--gpusall-p5005:5005Bert:最新--gpus卡号或所有查看卡号：nvidia-smi2，tf quota#自适应增长#config=tf。同胞。v1.ConfigProto（）#config。gpu选项。允许增长率
1、docker启动：
docker run --rm --gpus 0 -p5005:5005 bert:latest
docker run --rm --gpus all -p5005:5005 bert:latest
--gpus 卡号或者all
查看卡号：nvidia-smi
2、tf限额
# 自适应增长 # config = tf.compat.v1.ConfigProto() # config.gpu_options.allow_growth = True # sess = tf.compat.v1.Session(config=config) # 指定使用比例 config = tf.compat.v1.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.1 session = tf.compat.v1.Session(config=config)