i am sorry,I’m a beginner and don’t quite understand.
but i asked grok
he told me check
print(torch._path_)
print(torch.version.cuda)
print(torch.cuda.get_arch_list())
[‘C:\ProgramData\anaconda3\envs\yolov12\Lib\site-packages\torch’]
12.6
[‘sm_50’, ‘sm_60’, ‘sm_61’, ‘sm_70’, ‘sm_75’, ‘sm_80’, ‘sm_86’, ‘sm_90’]