CuPy (8.1.0) の (実験的な) AMD GPU サポートを使用/テストしようとしています。要するに:私はImportError
.
- ROCm を Ubuntu フォーカル マシン (gfx906/Radeon VII (Vega 20) カードを使用) にインストールする手順に従いました。
- ドキュメントから、前述の変数を(調整済みで
HCC_AMDGPU_TARGET
)ROCM_HOME
とを設定しましたexport PATH=$PATH:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin
。 pip3 install -v --no-cache-dir cupy
文句なく成功。-v
オプションのため、コンパイラに関する注意事項と警告がいくつかあります- 今、cupy をインポートしようとすると、
ImportError
.
In [1]: import cupy
---------------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
/usr/local/lib/python3.8/dist-packages/cupy/__init__.py in <module>
19 message='can\'t resolve package from __spec__')
---> 20 from cupy import core # NOQA
21 except ImportError as e:
/usr/local/lib/python3.8/dist-packages/cupy/core/__init__.py in <module>
----> 1 from cupy.core import core # NOQA
2 from cupy.core import internal # NOQA
3
cupy/core/core.pyx in init cupy.core.core()
cupy/core/_routines_manipulation.pyx in init cupy.core._routines_manipulation()
cupy/core/_routines_indexing.pyx in init cupy.core._routines_indexing()
cupy/core/_routines_math.pyx in init cupy.core._routines_math()
cupy/core/_reduction.pyx in init cupy.core._reduction()
cupy/core/_cub_reduction.pyx in init cupy.core._cub_reduction()
ModuleNotFoundError: No module named 'cupy.cuda.cub'
The above exception was the direct cause of the following exception:
ImportError Traceback (most recent call last)
<ipython-input-1-329ec5cf1bc8> in <module>
----> 1 import cupy
/usr/local/lib/python3.8/dist-packages/cupy/__init__.py in <module>
39 original error: {}'''.format(_exc_info[1])) # NOQA
40
---> 41 raise ImportError(_msg) from e
42
43
ImportError: CuPy is not correctly installed.
If you are using wheel distribution (cupy-cudaXX), make sure that the version of CuPy you installed matches with the version of CUDA on your host.
Also, confirm that only one CuPy package is installed:
$ pip freeze
If you are building CuPy from source, please check your environment, uninstall CuPy and reinstall it with:
$ pip install cupy --no-cache-dir -vvvv
Check the Installation Guide for details:
https://docs.cupy.dev/en/latest/install.html
original error: No module named 'cupy.cuda.cub'
次に何をすべきですか?コンパイルプロセスで何を探すべきですか? 前もって感謝します
編集ここで要求に応じて、インストールプロセスの完全な標準出力と標準エラー出力、およびコンソールのいくつかの io: https://seafile.cloud.uni-hannover.de/d/68862cd1057e47d180aa/