ckkelvinchan / BasicVSR_PlusPlus

Official repository of "BasicVSR++: Improving Video Super-Resolution with Enhanced Propagation and Alignment"
Apache License 2.0
568 stars 63 forks source link

Training: Why I must use cuda 10.1? #48

Open zqx1216155858 opened 6 months ago

zqx1216155858 commented 6 months ago

Traceback (most recent call last): File "/storage/public/home/2022124023/video-desnow/BasicVSR_PlusPlus/train.py", line 15, in from mmedit.apis import init_random_seed, set_random_seed, train_model File "/storage/public/home/2022124023/video-desnow/BasicVSR_PlusPlus/mmedit/apis/init.py", line 4, in from .matting_inference import init_model, matting_inference File "/storage/public/home/2022124023/video-desnow/BasicVSR_PlusPlus/mmedit/apis/matting_inference.py", line 8, in from mmedit.models import build_model File "/storage/public/home/2022124023/video-desnow/BasicVSR_PlusPlus/mmedit/models/init.py", line 2, in from .backbones import * # noqa: F401, F403 File "/storage/public/home/2022124023/video-desnow/BasicVSR_PlusPlus/mmedit/models/backbones/init.py", line 2, in from .sr_backbones import BasicVSRPlusPlus File "/storage/public/home/2022124023/video-desnow/BasicVSR_PlusPlus/mmedit/models/backbones/sr_backbones/init.py", line 3, in from .basicvsr_pp import BasicVSRPlusPlus File "/storage/public/home/2022124023/video-desnow/BasicVSR_PlusPlus/mmedit/models/backbones/sr_backbones/basicvsr_pp.py", line 6, in from mmcv.ops import ModulatedDeformConv2d, modulated_deform_conv2d File "/storage/public/home/2022124023/software/anaconda3/lib/python3.8/site-packages/mmcv/ops/init.py", line 2, in from .ball_query import ball_query File "/storage/public/home/2022124023/software/anaconda3/lib/python3.8/site-packages/mmcv/ops/ball_query.py", line 7, in ext_module = ext_loader.load_ext('_ext', ['ball_query_forward']) File "/storage/public/home/2022124023/software/anaconda3/lib/python3.8/site-packages/mmcv/utils/ext_loader.py", line 13, in load_ext ext = importlib.import_module('mmcv.' + name) File "/storage/public/home/2022124023/software/anaconda3/lib/python3.8/importlib/init.py", line 127, in import_module return _bootstrap._gcd_import(name[level:], package, level) ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory