start.sh 141 字节
Newer Older
0
0YuanZhang0 已提交
1
export FLAGS_fraction_of_gpu_memory_to_use=0.1
0
0YuanZhang0 已提交
2 3 4 5
port=$1
gpu=$2
export CUDA_VISIBLE_DEVICES=$gpu
python start_service.py ./infer_model $port 
0
0YuanZhang0 已提交
6