【报错】tensorflow_model_server: error while loading shared libraries: libcuda.so.1: cannot open shared

1、问题:

在docker中启动tensorflow/serving:1.12.0-gpu时,报错tensorflow_model_server: error while loading shared libraries: libcuda.so.1: cannot open shared

2、原因:

启动方式不对,创建容器时,添加参数:–runtime=nvidia

3、错误(官网上CPU版是这样启动的):

docker run -p 8500:8500 --mount type=bind,source=/data/anniegu_home/out_call/model/simnet_model,target=/models/simnet_model -e MODEL_NAME=simnet_model -t tensorflow/serving:1.12.0-gpu  &

正确:

docker run --runtime=nvidia -p 8500:8500 --mount type=bind,source=/data/anniegu_home/out_call/model/simnet_model,target=/models/simnet_model -e MODEL_NAME=simnet_model -t tensorflow/serving:1.12.0-gpu  &

版本安装等问题可以参考:
Docker tensorflow serving GPU libcuda.so.1问题

你可能感兴趣的:(报错)