layerism / TensorRT-Inference-Server-Tutorial

服务侧深度学习部署案例
453 stars 73 forks source link

when i use pre_process.py,i get a error #7

Closed 17702513221 closed 4 years ago

17702513221 commented 4 years ago

Traceback (most recent call last): File "pre_process.py", line 122, in export_path="../../model_repository" File "/usr/local/lib/python3.6/dist-packages/trtis-0.1.0-py3.6.egg/trtis/tf_backend/tf2graphdef.py", line 68, in tf2graphdef File "/home/xs/.local/lib/python3.6/site-packages/tensorflow_core/python/client/session.py", line 1585, in init super(Session, self).init(target, graph, config=config) File "/home/xs/.local/lib/python3.6/site-packages/tensorflow_core/python/client/session.py", line 699, in init self._session = tf_session.TF_NewSessionRef(self._graph._c_graph, opts) tensorflow.python.framework.errors_impl.InvalidArgumentError: Invalid device ordinal value (1). Valid range is [0, 0]. while setting up XLA_GPU_JIT device number 1