arm64架构jetson，torch检测不到gpu

本人小白，如有不对欢迎批评，确实是自己解决问题花了一整天，所以写个文章记录以下，如果能帮到别人那就更好了

问题 宿主机torch版本对应了cuda11.4，却根本找不到gpu，在csdn上找文章，

https://blog.youkuaiyun.com/qq_37700257/article/details/128645779?ops_request_misc=%257B%2522request%255Fid%2522%253A%252204d099fa8fbb9d33f44ae06a44b9ad28%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=04d099fa8fbb9d33f44ae06a44b9ad28&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~top_positive~default-1-128645779-null-null.142^v100^pc_search_result_base1&utm_term=jetson%20torch&spm=1018.2226.3001.4187

PyTorch for Jetson - Jetson & Embedded Systems / Announcements - NVIDIA Developer Forums

在这里找自己cuda对应版本的arm64架构torch，下载了之后传到宿主机进行下载，结束之后查看版本

torch                  2.0.0+nv23.5
 

是你对应版本的torch然后查看能否调用到gpu