如何解决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中设置上述变量(调整后的
ImportError
,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
选项,仅提供了一些编译器注释和警告 - 现在,当我尝试导入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
没有建立。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。