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