问题遇到的现象和发生背景
使用python训练模型时报错:ValueError: The 'astra_cuda' impl
is not found. This impl
is either not supported, it may be misspelled, or external packages required are not available. Consult RAY_TRAFO_IMPLS
to find the run-time available implementations.
问题相关代码,请勿粘贴截图
复现https://github.com/hongwang01/InDuDoNet
运行结果及报错内容
Traceback (most recent call last):
File "train.py", line 23, in <module>
from deeplesion.Dataset import MARTrainDataset
File "/root/IDD+/deeplesion/__init__.py", line 1, in <module>
from .Dataset import MARTrainDataset, nmar_prior, nmarprior
File "/root/IDD+/deeplesion/Dataset.py", line 18, in <module>
ray_trafo = build_gemotry(param)
File "/root/IDD+/deeplesion/build_gemotry.py", line 75, in build_gemotry
ray_trafo_hh = odl.tomo.RayTransform(reco_space_h, geometry_h, impl='astra_cuda')
File "/root/miniconda3/envs/my-env/lib/python3.6/site-packages/odl/tomo/operators/ray_trafo.py", line 193, in __init__
impl_type, self.__cached_impl = self._initialize_impl(impl)
File "/root/miniconda3/envs/my-env/lib/python3.6/site-packages/odl/tomo/operators/ray_trafo.py", line 238, in _initialize_impl
'implementations.'.format(impl)
ValueError: The 'astra_cuda' `impl` is not found. This `impl` is either not supported, it may be misspelled, or external packages required are not available. Consult `RAY_TRAFO_IMPLS` to find the run-time available implementations.
我的解答思路和尝试过的方法
尝试下载astra-toolbox,但没成功
我想要达到的结果
顺利进行训练