I replace the code with yours, but I got the output as this![result](https://user-images.githubusercontent.com/53261018/186422373-0678c37a-8d2d-45ac-97cd-b772c99f0368.jpg)

Open cuixing100876 opened this issue 2 years ago • 2 comments

          I replace the code with yours, but I got the output as this![result](https://user-images.githubusercontent.com/53261018/186422373-0678c37a-8d2d-45ac-97cd-b772c99f0368.jpg)

Originally posted by @phhandong in https://github.com/diyiiyiii/StyTR-2/issues/10#issuecomment-1225680160

Jan 25 '24 02:01 cuixing100876

I have the same problem as @phhandong

Jan 25 '24 02:01 cuixing100876

I guess it's a CUDA version mismatch that's causing this problem, I solve this problem by switching my device to V100 with lower CUDA version, you can try is my way.

Jan 25 '24 02:01 phhandong