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

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

New Issue
Closed
opened 2026-01-29 21:35:55 +00:00 by claunia · 2 comments
claunia commented 2026-01-29 21:35:55 +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)
claunia commented 2026-01-29 21:35:59 +00:00
Author
Owner
Copy Link

@xinntao commented on GitHub (Jun 23, 2021):

I have not encountered this issue.
May check the torchvision ? （the version in my env is torchvision 0.8.2）

@xinntao commented on GitHub (Jun 23, 2021): I have not encountered this issue. May check the torchvision ? （the version in my env is torchvision 0.8.2）
claunia commented 2026-01-29 21:36:02 +00:00
Author
Owner
Copy Link

@lxy2017 commented on GitHub (Jun 24, 2021):

I have not encountered this issue.
May check the torchvision ? （the version in my env is torchvision 0.8.2）

Thanks!
It works for torchvision 0.8.2

@lxy2017 commented on GitHub (Jun 24, 2021): > I have not encountered this issue. > May check the torchvision ? （the version in my env is torchvision 0.8.2） Thanks! It works for torchvision 0.8.2
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#8
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?