2
AI人工智能
怎样使用DCU加速训练？
2021-11-25 16:17:40
309

attachmentId-image.png您好，我根据教程在E-Shell中安装miniconda和pytorch，用pytorch跑深度学习训练的时候，提示没有可用CUDA，只能cpu训练，请问怎样操作可以用DCU加速训练呢？

全部评论（5）
只看楼主
G
ghdevadmin
1楼

rocm-smi看下GPU占用有显示吗？请注意要在计算节点运行程序，login节点没有加速卡

2021/11/25
A
ac490q7j1h
回复 @ghdevadmin

感谢回复，请问怎么在计算节点运行程序呢，是我下图这种操作吗？
watch rocm-smi一直没有占用显示

2021/11/25
T
tianlh
回复 @ghdevadmin

这种方式可以，这是申请到了计算节点，ssh过来正常运行程序就可以了，也可以使用slurm作业提交方式，请参考https://developer.hpccube.com/wp-content/uploads/2021/07/SLURM%E4%BB%8B%E7%BB%8D%E4%B8%8E%E4%BD%9C%E4%B8%9A%E6%8F%90%E4%BA%A4.pdf

2021/11/25
A
ac490q7j1h
回复 @ghdevadmin

您好，请帮忙看下是不是还没有dcu加速成功啊，脚本文件按下面的设置的，然后sbatch 脚本运行，gpu_mem一直是0，是哪里出错了吗

2021/12/1
全部4条回复
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据
常见问题新帖
暂无数据