框架组件
加载pytorch出现HIP error
2024-03-19 09:42:47
214

出现了如下问题

attachmentId-image.png

之前曾经出现过这个问题,有老师回复称这个版本有点老了,对应的miopen也是旧版本,换用以下命令解决了

module switch compiler/rocm/4.0.1

module load apps/PyTorch/1.7.0a0/hpcx-2.4.1-gcc-7.3.1-rocm4.0.1

最近仍使用以上命令,但又出现以上问题,是否是需要进一步更新版本?

全部评论(3)
只看楼主
L
louis
1楼

报错信息显示为未找到可用设备,请您先确认一下运行的节点是否为login节点,login节点没有配置DCU卡

2021/12/27
G
ghfund_i_b5
回复 @louis

是通过提交作业的方式运行的
#!/bin/bash
#SBATCH -J test
#SBATCH -p kshdtest
#SBATCH -N 1
#SBATCH --ntasks-per-node 4
#SBATCH --cpus-per-task=8

module switch compiler/rocm/4.0.1
module load apps/PyTorch/1.7.0a0/hpcx-2.4.1-gcc-7.3.1-rocm4.0.1
#python3 test.py
python3 /public/home/ghfund_i_b5/slurm_template/GAN.py

2021/12/28
T
tangxd
回复 @louis

#SBATCH --gres=dcu:4 需加上这个参数 申请DCU资源

2021/12/30
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据