-- Process 1 terminated with the following error:
Traceback (most recent call last):
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/torch/multipr
fn(i, args)
File "/home/zkl/code/TPVFormer/train.py", line 75, in main
from builder import tpv_lidarseg_builder as model_builder
File "/home/zkl/code/TPVFormer/builder/tpv_lidarseg_builder.py", line 1, in
from tpvformer10 import
File "/home/zkl/code/TPVFormer/tpvformer10/init.py", line 2, in
from .modules import *
File "/home/zkl/code/TPVFormer/tpvformer10/modules/init.py", line 1, in
from .image_cross_attention import TPVImageCrossAttention, TPVMSDeformableAttention3D
File "/home/zkl/code/TPVFormer/tpvformer10/modules/image_cross_attention.py", line 2, in
from mmcv.ops.multi_scale_deform_attn import multi_scale_deformable_attn_pytorch
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/mmcv/ops/in
from .assign_score_withk import assign_score_withk
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/mmcv/ops/assi
ext_module = ext_loader.load_ext(
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/mmcv/utils/ex
ext = importlib.import_module('mmcv.' + name)
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/importlib/init__.py", lin
return _bootstrap._gcd_import(name[level:], package, level)
ImportError: /opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/mmcv/_exK2at6Tensor7is_cudaEv
(py38_cu111_tpvformer_tesla_zkl) :~/code/TPVFormer$ bash launcher.sh config/tpv_lid Namespace(gpus=2, py_config='config/tpv_lidarseg.py', resume_from='', work_dir='out/tpv_lidarseg') tcp://127.0.0.1:20506 tcp://127.0.0.1:20506 Traceback (most recent call last): File "train.py", line 329, in
torch.multiprocessing.spawn(main, args=(args,), nprocs=args.gpus)
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/torch/multipr
return start_processes(fn, args, nprocs, join, daemon, start_method='spawn')
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/torch/multipr
while not context.join():
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/torch/multipr
raise ProcessRaisedException(msg, error_index, failed_process.pid)
torch.multiprocessing.spawn.ProcessRaisedException:
-- Process 1 terminated with the following error: Traceback (most recent call last): File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/torch/multipr fn(i, args) File "/home/zkl/code/TPVFormer/train.py", line 75, in main from builder import tpv_lidarseg_builder as model_builder File "/home/zkl/code/TPVFormer/builder/tpv_lidarseg_builder.py", line 1, in
from tpvformer10 import
File "/home/zkl/code/TPVFormer/tpvformer10/init.py", line 2, in
from .modules import *
File "/home/zkl/code/TPVFormer/tpvformer10/modules/init.py", line 1, in
from .image_cross_attention import TPVImageCrossAttention, TPVMSDeformableAttention3D
File "/home/zkl/code/TPVFormer/tpvformer10/modules/image_cross_attention.py", line 2, in
from mmcv.ops.multi_scale_deform_attn import multi_scale_deformable_attn_pytorch
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/mmcv/ops/in
from .assign_score_withk import assign_score_withk
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/mmcv/ops/assi
ext_module = ext_loader.load_ext(
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/mmcv/utils/ex
ext = importlib.import_module('mmcv.' + name)
File "/opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/importlib/init__.py", lin
return _bootstrap._gcd_import(name[level:], package, level)
ImportError: /opt/anaconda3/envs/py38_cu111_tpvformer_tesla_zkl/lib/python3.8/site-packages/mmcv/_exK2at6Tensor7is_cudaEv