TencentARC/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2026-02-15 13:54:33 +00:00
Code Issues 476 Packages Projects Releases 8 Wiki Activity

RuntimeError: Could not run 'torchvision::roi_align' with arguments from the 'CUDA' backend. #5

New Issue
Open
opened 2026-01-29 21:35:52 +00:00 by claunia · 0 comments
claunia commented 2026-01-29 21:35:52 +00:00
Owner
Copy Link

Originally created by @lxy2017 on GitHub (Jun 21, 2021).

When I train the code, I get the following error

RuntimeError: Could not run 'torchvision::roi_align' with arguments from the 'CUDA' backend. 'torchvision::roi_align' is only available for these backends: [CPU, BackendSelect, Named, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradPrivateUse1, AutogradPrivateUse2, AutogradPrivateUse3, Tracer, Autocast, Batched, VmapMode].

image

Originally created by @lxy2017 on GitHub (Jun 21, 2021). When I train the code, I get the following error RuntimeError: Could not run 'torchvision::roi_align' with arguments from the 'CUDA' backend. 'torchvision::roi_align' is only available for these backends: [CPU, BackendSelect, Named, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradPrivateUse1, AutogradPrivateUse2, AutogradPrivateUse3, Tracer, Autocast, Batched, VmapMode]. ![image](https://user-images.githubusercontent.com/9693098/122701302-7de87d00-d27f-11eb-8376-77149695cc22.png)
Sign in to join this conversation.
No Branch/Tag Specified
Branches Tags
Labels
Clear labels
pull-request
Mirrored from GitHub Pull Request
No Label
Milestone
No items
No Milestone
Projects
Clear projects
No project
Assignees
Clear assignees
claunia
No Assignees
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: TencentARC/GFPGAN#5
Delete Branch "%!s()"

Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?