- 博客(2)
- 收藏
- 关注
RSS订阅原创 arm64架构jetson，torch检测不到gpu
本人小白，如有不对欢迎批评，确实是自己解决问题花了一整天，所以写个文章记录以下，如果能帮到别人那就更好了。在这里找自己cuda对应版本的arm64架构torch，下载了之后传到宿主机进行下载，结束之后查看版本。问题 宿主机torch版本对应了cuda11.4，却根本找不到gpu，在csdn上找文章，是你对应版本的torch然后查看能否调用到gpu。
2024-12-18 16:30:19
300
原创 jetson设备，arm64架构，问题解决＞＞＞ import torchvision/home/nvidia/.local/lib/python3.8/site-packages/torchvis
这里有一个小插曲，楼主的显示是0.15.1，原因是有多版本的torchvision，直接使用pip uninstall torchvision将pip的删除掉，上面的编译到了系统级，默认就使用那个了。这里的原因是nvidia官方提供了arm64架构的torch，普通下载的torch调取不到gpu（可能是我个人问题，本人也是小白，正在学习）这几个torch包都是很奇怪的版本，torch在上面的文章解决之后，torchvision也需要自己编译下载才会是这个版本，切换到你的torch对应的版本。
2024-12-18 16:22:59
548
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人