Paddle3D icon indicating copy to clipboard operation
Paddle3D copied to clipboard

训练CaDDN模型时,编译grid_sample_3d算子失败!

Open melonyxin opened this issue 2 years ago • 3 comments

966c2c95cf9b67ad2ce3631bedb3247 在国产显卡DCU环境上训练CaDDN模型出现上图报错,查看了一下grid_sample_3d是cuda c写的算子,是因为DCU是ROCm框架,没有cuda环境,所以编译失败吗? CaDDN是否支持在ROCm环境下进行训练呢?

melonyxin avatar Aug 22 '22 08:08 melonyxin

paddle3d中的grid_sample_3d是cuda c写的算子,暂时不支持ROCm框架。 目前develop版的paddle已经支持grid_sample_3d算子了,您可以安装develop版的paddle,删去paddle3d中自定义grid_sample_3d的代码,用paddle自带的grid_sample_3d试试看~

wangna11BD avatar Aug 29 '22 03:08 wangna11BD

paddle3d中的grid_sample_3d是cuda c写的算子,暂时不支持ROCm框架。 目前develop版的paddle已经支持grid_sample_3d算子了,您可以安装develop版的paddle,删去paddle3d中自定义grid_sample_3d的代码,用paddle自带的grid_sample_3d试试看~

您好,我在develop版本的api里没有中找到grid_sample_3d算子,能帮忙发现链接,学习下使用方法吗?

sunhm15 avatar Sep 05 '22 02:09 sunhm15

https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/api/paddle/nn/functional/grid_sample_cn.html#canshu develop版本的grid_sample支持5d输入,5d输入就是grid_sample_3d

wangna11BD avatar Sep 05 '22 06:09 wangna11BD