Closed adamm123 closed 4 years ago
Yes Tritonserver 20.03 does indeed use Pytorch 1.5.0 from nvcr.io/nvidia/pytorch:20.03-py3. The docs will be fixed to reflect the same. Thank you for pointing that out.
I would recommend using the Tritonserver 20.01 (or 19.12) container since they are based on Pytorch 1.4.0 and libtorch is known to have brittle backward compatibility between some versions. It is likely the op in question here was modified or removed in 1.5.0
Description A clear and concise description of what the bug is.
Cannot load FasterRCNN model exported to TorchScript using Triton r20.03.
Triton Information What version of Triton are you using?
nvcr.io/nvidia/tritonserver:20.03-py3
Are you using the Triton container or did you build it yourself?
container
To Reproduce Steps to reproduce the behavior.
Describe the models (framework, inputs, outputs), ideally include the model configuration file (if using an ensemble include the model configuration file for that as well).
ENVIRONMENT:
model = torchvision.models.detection.fasterrcnn_resnet50_fpn() model = model.eval() script = torch.jit.script(model) script.save('model.pt')
mkdir -p models/object_detection/1 cp model.pt models/object_detection/1
nvidia-docker run --rm --shm-size=1g --ulimit memlock=-1 --ulimit stack=67108864 -p8000:8000 -p8001:8001 -p8002:8002 -v//models:/models nvcr.io/nvidia/tritonserver:20.03-py3 trtserver --model-repository=/models --log-verbose=1 --strict-model-config=false
model_repository_manager.cc:840] failed to load 'object_detection' version 1: Internal: load failed for libtorch model -> 'object_detection': Unknown builtin op: torchvision::_new_empty_tensor_op. Could not find any similar ops to torchvision::_new_empty_tensor_op. This op may not exist or may not be currently supported in TorchScript.