I have the following error when i run this model. Can you tell me which pytorch version to use to run this model.
I have the following error
/home/nvidia/.local/lib/python3.8/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")
Traceback (most recent call last):
File "test.py", line 3, in
from evaluation.eval_wrapper import eval_lane
File "/home/nvidia/Ultra-Fast-Lane-Detection-v2/evaluation/eval_wrapper.py", line 2, in
from data.dataloader import get_test_loader
File "/home/nvidia/Ultra-Fast-Lane-Detection-v2/data/dataloader.py", line 4, in
import torchvision.transforms as transforms
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/init.py", line 5, in
from torchvision import datasets, io, models, ops, transforms, utils
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/models/init.py", line 17, in
from . import detection, optical_flow, quantization, segmentation, video
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/models/quantization/init.py", line 3, in
from .mobilenet import
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/models/quantization/mobilenet.py", line 1, in
from .mobilenetv2 import # noqa: F401, F403
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/models/quantization/mobilenetv2.py", line 5, in
from torch.ao.quantization import DeQuantStub, QuantStub
ImportError: cannot import name 'QuantStub' from 'torch.ao.quantization' (/home/nvidia/.local/lib/python3.8/site-packages/torch/ao/quantization/init.py)
I have the following error when i run this model. Can you tell me which pytorch version to use to run this model.
I have the following error
/home/nvidia/.local/lib/python3.8/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: warn(f"Failed to load image Python extension: {e}") Traceback (most recent call last): File "test.py", line 3, in
from evaluation.eval_wrapper import eval_lane
File "/home/nvidia/Ultra-Fast-Lane-Detection-v2/evaluation/eval_wrapper.py", line 2, in
from data.dataloader import get_test_loader
File "/home/nvidia/Ultra-Fast-Lane-Detection-v2/data/dataloader.py", line 4, in
import torchvision.transforms as transforms
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/init.py", line 5, in
from torchvision import datasets, io, models, ops, transforms, utils
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/models/init.py", line 17, in
from . import detection, optical_flow, quantization, segmentation, video
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/models/quantization/init.py", line 3, in
from .mobilenet import
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/models/quantization/mobilenet.py", line 1, in
from .mobilenetv2 import # noqa: F401, F403
File "/home/nvidia/.local/lib/python3.8/site-packages/torchvision/models/quantization/mobilenetv2.py", line 5, in
from torch.ao.quantization import DeQuantStub, QuantStub
ImportError: cannot import name 'QuantStub' from 'torch.ao.quantization' (/home/nvidia/.local/lib/python3.8/site-packages/torch/ao/quantization/init.py)