sugarme / gotch

Go binding for Pytorch C++ API (libtorch)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

bugfix cuda_device_cnt

ZonghaoWang opened this issue · comments

image
for atc_cuda_device_count return int32 value, so the code should be change to
image

@ZonghaoWang ,

Thanks for the report!