Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

SortPairsDescending to calculate temp_storage_bytes, status: invalid device function #429

Open
juelietq opened this issue Dec 18, 2023 · 1 comment
Assignees

Comments

@juelietq
Copy link

本地环境如下:GPU Compute Capability: 6.1, Driver API Version: 11.2, Runtime API Version: 11.2,cuDNN Version: 8.1,paddlepaddle版本:2.5.2.post112,paddle3D成功训练,但evaluate评估时报错: top_k_function_cuda.h:1131] TopKOP failed as could not launch cub::DeviceSegmentedRadixSort::SortPairsDescending to calculate temp_storage_bytes, status: invalid device function

@wangna11BD
Copy link
Collaborator

wangna11BD commented Jan 18, 2024

请问具体是跑的哪个模型呢,evaluate命令是什么呢,报错信息可以给个完整的截图吗

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants