本人是第一次使用jeston nano,下载torch和torchvision碰了很多壁,因此这里把经验教训写下来,希望能帮助到像我一样的小白,让大家少走点弯路,由于第一次在csdn编辑,格式不美观,有错误见谅~
方式一:通过离线包安装
1.安装jtop,查看jetpack版本
pip3 install jetson-stats
安装好jtop后,输入:
jtop
在图中可以看到我的jetpack版本是 4.2,cuda版本是10.0(cuda的安装在网上有许多教程),后续安装torch和torchvision需要根据这个jetpack版本和python版本来,否则会报错
OSError: libcudart.so.10.0: cannot open shared object file: No such file
在英伟达官网查找jetpack对应torch版本,以下贴出我在官网查询的结果