jetsonnano安装torch1.8.0与torchvision0.9.0（个人血泪史）

KongTiaoXuLun

已于 2023-05-23 18:44:04 修改

阅读量2.6k

点赞数 6

文章标签： python 开发语言

于 2023-05-23 18:42:26 首次发布

本文链接：https://blog.youkuaiyun.com/cyh20182808/article/details/130832990

版权

该文详细记录了在JetsonNano上部署YOLOv5并使用TensorRTX加速的过程，包括遇到的CUDA安装错误及解决方法。作者首先介绍了所需的CUDA版本和PyTorchtorchvision版本，然后详述了安装步骤，特别指出在安装torchvision时需安装依赖库并修改环境变量以解决编译错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考链接：Jetson Nano部署YOLOv5与Tensorrtx加速——（自己走一遍全过程记录）_jetson nano yolov5_Mr_LanGX的博客-优快云博客

jetson 装机问题简记 - 知乎 (zhihu.com)

error: command ‘/usr/local/cuda:/usr/local/cuda/bin/nvcc‘ failed with exit status 1_小杨同学的技术博客_51CTO博客

我的板板是4.6，cuda10.2

我要安装的是1.8.0的torch与对应0.9.0的torchvision

下载链接：pytorch-torchvison.zip_免费高速下载|百度网盘-分享无限制 (baidu.com)

提取码yvex

里面有一个whl文件的torch，与一个torchvision的文件夹

那么把这两个传到板板上，最好建一个文件夹

（我已经安好了archiconda3，虚拟环境python3.6）

然后在板板的命令端，进入这个文件夹中，命令就是

cd 文件夹名

安装whl文件简单，直接

pip install torch-1.8.0-cp36-cp36m-linux_aarch64.whl

安装torchvision，先安装一堆依赖

sudo apt-get install libopenmpi2
sudo apt-get install libopenblas-dev
sudo apt-get install libjpeg-dev zlib1g-dev

要再进入torchvision这个文件夹

cd torchvision

再输入

export BUILD_VERSION=0.9.0

注意了，这里我输入这条命令才好使（加了sudo的不好使，说我非法指令，怎么改都不行）

python3 setup.py install

然后呢，开始编译了，那么最后可能会报错，例如我就报了这个

error: command ‘/usr/local/cuda:/usr/local/cuda/bin/nvcc‘ failed with exit status 1

解决:在命令端输入

sudo gedit ~/.bashrc

然后在弹出来这个小笔记中呢，找到

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

给它改成export CUDA_HOME=:/usr/local/cuda

如我的图改完了这是

然后右上角save以下，然后在命令行输入

source ~/.bashrc

然后重启了一下，重新来一遍咱们的操作，从进torchvision这个文件夹开始，然后就，就安装成功了，我吃完饭回来看到成了，高兴的想给它磕两个，安了好多天，各种各样的torchvision就是怎么都安不上，好了不叭叭了