Open
Description
Bug Description
api/test_classes.py:6: in <module>
import torch_tensorrt as torchtrt
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/__init__.py:125: in <module>
from torch_tensorrt.runtime import * # noqa: F403
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/runtime/__init__.py:1: in <module>
from torch_tensorrt.dynamo.runtime import ( # noqa: F401
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/__init__.py:10: in <module>
from ._compiler import (
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/_compiler.py:16: in <module>
from torch_tensorrt.dynamo import _defaults, partitioning
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/partitioning/__init__.py:1: in <module>
from ._adjacency_partitioner import partition as fast_partition
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/partitioning/_adjacency_partitioner.py:20: in <module>
from torch_tensorrt.dynamo.conversion._ConverterRegistry import (
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/conversion/__init__.py:1: in <module>
from . import (
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/conversion/custom_ops_converters.py:15: in <module>
from torch_tensorrt.dynamo.lowering.passes.fuse_distributed_ops import (
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/lowering/__init__.py:7: in <module>
from .passes import *
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/lowering/passes/__init__.py:1: in <module>
from ._aten_lowering_pass import *
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/lowering/passes/_aten_lowering_pass.py:31: in <module>
if not is_tegra_platform():
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch_tensorrt/dynamo/utils.py:817: in is_tegra_platform
if torch.cuda.get_device_capability() in [(8, 7), (7, 2)]:
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch/cuda/__init__.py:571: in get_device_capability
prop = get_device_properties(device)
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch/cuda/__init__.py:587: in get_device_properties
_lazy_init() # will define _get_device_properties
/opt/python/cp39-cp39/lib/python3.9/site-packages/torch/cuda/__init__.py:383: in _lazy_init
torch._C._cuda_init()
E RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
To Reproduce
Steps to reproduce the behavior:
Expected behavior
Environment
Build information about Torch-TensorRT can be found by turning on debug messages
- Torch-TensorRT Version (e.g. 1.0.0):
- PyTorch Version (e.g. 1.0):
- CPU Architecture:
- OS (e.g., Linux):
- How you installed PyTorch (
conda,pip,libtorch, source): - Build command you used (if compiling from source):
- Are you using local sources or building from archives:
- Python version:
- CUDA version:
- GPU models and configuration:
- Any other relevant information: