1.测试环境:
openeuler-2403系统自带python3.11.6版本
sudo tar cvjf DTK-24.04.1-openEuler22.03-x86_64.tar -C /opt
sudo ./rock-5.7.1-6.2.26-V1.5.aio.run
flavr_pytorch 参考: https://sourcefind.cn/#/model-zoo/1728023579688325121
pip3 install torch-2.1.0+das1.1.git3ac1bdd.abi1.dtk2404-cp311-cp311-manylinux_2_31_x86_64.whl
pip3 install torchvision-0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1-cp311-cp311-manylinux_2_31_x86_64.whl
2.报错如下:
cd flavr_pytorch
$ python3 main.py --batch_size 32 --test_batch_size 32 --dataset vimeo90K_septuplet --loss 1*L1 --max_epoch 200 --lr 0.0002 --data_root datasets --n_outputs 1 --num_gpu 1
Namespace(dataset='vimeo90K_septuplet', data_root='datasets', model='unet_18', nbr_frame=4, nbr_width=1, joinType='concat', upmode='transpose', n_outputs=1, loss='1*L1', lr=0.0002, beta1=0.9, beta2=0.99, batch_size=32, test_batch_size=32, start_epoch=0, max_epoch=200, resume=False, resume_exp=None, checkpoint_dir='.', load_from=None, pretrained=None, exp_name='exp', log_iter=60, num_gpu=1, random_seed=12345, num_workers=16, use_tensorboard=False, val_freq=1, cuda=True)
Building model: unet_18
open comgr lib:libamd_comgr.so error
load comgr library error
Traceback (most recent call last):
File "/home/iscas/flavr_pytorch/main.py", line 66, in <module>
model = torch.nn.DataParallel(model).to(device)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/iscas/.local/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1160, in to
return self._apply(convert)
^^^^^^^^^^^^^^^^^^^^
File "/home/iscas/.local/lib/python3.11/site-packages/torch/nn/modules/module.py", line 810, in _apply
module._apply(fn)
File "/home/iscas/.local/lib/python3.11/site-packages/torch/nn/modules/module.py", line 810, in _apply
module._apply(fn)
File "/home/iscas/.local/lib/python3.11/site-packages/torch/nn/modules/module.py", line 810, in _apply
module._apply(fn)
[Previous line repeated 1 more time]
File "/home/iscas/.local/lib/python3.11/site-packages/torch/nn/modules/module.py", line 833, in _apply
param_applied = fn(param)
^^^^^^^^^
File "/home/iscas/.local/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1158, in convert
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/iscas/.local/lib/python3.11/site-packages/torch/cuda/__init__.py", line 298, in _lazy_init
torch._C._cuda_init()
RuntimeError: No HIP GPUs are available

暂无数据
cd /opt/dtk-24.04.1/lib64
ls -l libamd_comgr.so*
lrwxrwxrwx. 1 root root 17 Jun 13 2024 libamd_comgr.so -> libamd_comgr.so.2
lrwxrwxrwx. 1 root root 27 Jun 13 2024 libamd_comgr.so.2 -> libamd_comgr.so.2.4.24.04.1
-rwxr-xr-x. 1 root root 133057208 Jun 9 2024 libamd_comgr.so.2.4.24.04.1