第一个坑
安装官网教程直接安装 pip3 install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/cu118
不管什么版本,都用不了cuda
找到nvidia官网论坛上有专门的介绍jetson板子的驱动:PyTorch for Jetson – Jetson & Embedded Systems / Jetson Nano – NVIDIA Developer Forums
然后下载whl后直接安装,下载文件名比较长
第二个坑 文件名不能修改
修改之后安装的时候会报错,报的就是文件名不对,也不知道这个设计思路到底是为啥
安装完torch之后,可以看到这个版本比较特殊,好歹cuda可以用了
使用yolo的时候下一个坑出现了,ultralytics 需要torchvision模块,想当然的就 pip install torchvision,结果把刚安装的torch给卸载了,重新装上了torch 2.1.0版本,一夜回到解放前
只能找jetson板子对应的torch vision模块,找了半天,竟然在刚才下载torch的页面上,而且还贴心的给出了torch和torchvision版本对应关系和安装方法,真的感动死