This seems to be a common issue with 3090 for many projects:
NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75.
If you want to use the NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
Traceback (most recent call last):
File "demo_video.py", line 62, in
Fire(main)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/fire/core.py", line 138, in Fire
component_trace = _Fire(component, args, parsed_flag_args, context, name)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/fire/core.py", line 468, in _Fire
target=component.name)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/fire/core.py", line 672, in _CallAndUpdateTrace
component = fn(*varargs, **kwargs)
File "demo_video.py", line 55, in main
tracked_bboxes = track(tracker, video, initial_bbox)
File "demo_video.py", line 24, in track
tracker.initialize(frames[0], initial_bbox)
File "/home/dualeco/Documents/FEARTracker-main/model_training/tracker/fear_tracker.py", line 34, in initialize
self._template_features = self.get_template_features(image, rect)
File "/home/dualeco/Documents/FEARTracker-main/model_training/tracker/fear_tracker.py", line 44, in get_template_features
return self.net.get_features(img)
File "/home/dualeco/Documents/FEARTracker-main/model_training/model/fear_net.py", line 64, in get_features
features = self.feature_extractor(crop)
File "/home/dualeco/Documents/FEARTracker-main/model_training/model/fear_net.py", line 60, in feature_extractor
x = stage(x)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/torch/nn/modules/container.py", line 117, in forward
input = module(input)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/dualeco/Documents/FEARTracker-main/src/mobile-cv/mobile_cv/arch/fbnet_v2/basic_blocks.py", line 292, in forward
x = self.conv(x)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 423, in forward
return self._conv_forward(input, self.weight)
File "/home/dualeco/.conda/envs/py37fear/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 420, in _conv_forward
self.padding, self.dilation, self.groups)
RuntimeError: CUDA error: no kernel image is available for execution on the device
PinataFarms
