PyTorch Forums
Custom CUDA operator only work well on cuda:0
C++
shzygmyx
(Yuxian Meng)
May 2, 2022, 3:06pm
7
It works! Thank you so much!
2 Likes
show post in topic