AMD GPU上的CuPy导致ImportError

问题描述

我正在尝试使用/测试CuPy(8.1.0)的(实验)AMD GPU支持。简而言之:我有一个const [upload,setUpload] = useState([]); const handleSubmit = (e) => { e.preventDefault(); var formData = new FormData(); formData.append('upload',upload); ... } ... <Form.Control type="file" className="form-control" name="upload" onChange={(e) => setUpload(e.target.files)} multiple />

  • 我按照说明在Ubuntu聚焦机上安装ROCm(使用gfx906 / Radeon VII(Vega 20)卡)。
  • doc中设置上述变量(调整后的ImportErrorHCC_AMDGPU_TARGETROCM_HOME
  • export PATH=$PATH:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin成功了,没有任何抱怨。由于pip3 install -v --no-cache-dir cupy选项,仅提供了一些编译器注释和警告
  • 现在,当我尝试导入Cupy时,它会抛出-v
ImportError

下一步我该怎么做?在编译过程中我应该寻找什么?预先感谢

编辑,应此处要求,提供安装过程的完整标准输出和标准错误以及控制台的一些输入输出https://seafile.cloud.uni-hannover.de/d/68862cd1057e47d180aa/

解决方法

总结到目前为止的评论中的讨论:

  • 对于CuPy v8.x(最新的稳定版本):在v8.1之前的ROCm / HIP环境中未构建有问题的cupy.cuda.cub模块,希望该模块将在v8.2中修复(请参见{{ 3}})。
  • 对于CuPy v9.x(主分支):只要正确安装了rocPRIM和hipCUB,它就应该可以工作。可能是由于OP的本地环境设置不正确,导致cupy.cuda.cub模块未构建,因此ImportError没有建立。

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...