yolov5--nvidia xavier的conda环境配置--pytorch

yolov5–nvidia xavier的conda环境配置–pytorch

本机的cuda为10.2，安装1.7.0的torch与0.8.1的torchvision

1.arm版本的conda-miniforge的安装

在arm架构的推理板中，无法安装anaconda，所以选用Mambaforge进行环境隔离。Mambaforge的使用与anaconda的使用方法一致。

miniforge与miniconda的区别在于miniforge的下载通道是conda-forge
miniforge的github地址：https://github.com/conda-forge/miniforge
miniforge的.sh release下载地址：https://github.com/conda-forge/miniforge/releases

1.1. 安装miniforge

通过执行如下命令进行安装，安装完成后，就和anaconda的使用方式一样。

sh Mambaforge-4.9.2-5-Linux-aarch64.sh

1.2 创建虚拟环境yolov5_env
创建新的虚拟环境，环境名字是yolov5_env,由于使用python3.6，所以在创建虚拟环境时，拉入3.6的python。执行以下命令进行环境创建：

conda create -n yolov5_env python=3.6

1.3 安装yolov5环境支持
以下为yolov5的requirements：

# pip install -r requirements.txt

# base ----------------------------------------
Cython
matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow
PyYAML>=5.3
scipy>=1.4.1
tensorboard>=2.2
torch>=1.7.0
torchvision>=0.8.1
tqdm>=4.41.0

# logging -------------------------------------
# wandb

# plotting ------------------------------------
seaborn>=0.11.0
pandas

# export --------------------------------------
# coremltools==4.0
# onnx>=1.8.0
# scikit-learn==0.19.2  # for coreml quantization

# extras --------------------------------------
thop  # FLOPS computation
pycocotools>=2.0  # COCO mAP

在进行依赖安装的过程中，建议一个一个使用conda install安装，不然可能依赖会从系统python的的site-packages里面调，下面重点讲一下torch以及torchvision的安装。

2.安装torch与torchvision

2.1 torch的安装

jetson的torch下载地址为：

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-7-0-now-available/72048

根据板子的jetpack版本选择对应的torch版本，我是用的是jetpack4.4，所以下载了如下版本：

PyTorch v1.7.0

JetPack 4.4（L4T R32.4.3）/ JetPack 4.4.1（L4T R32.4.4）/ JetPack 4.5（L4T R32.5.0）
Python 3.6- torch-1.7.0-cp36-cp36m-linux_aarch64.whl 947

下载完后，执行以下命令：

conda activate yolov5_env
sudo apt-get update
sudo apt-get install libopenmpi2
sudo apt-get install libopenblas-dev

进入虚拟环境后执行：

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

如果不放心可以执行以下命令：

/home/nvidia/mambaforge/envs/yolov5_env/bin/pip install torch-1.7.0-cp36-cp36m-linux_aarch64.whl

这样直接确保是虚拟环境内的pip安装torch，接下来就是耐心的等待安装完成即可。
2.2安装torchvision
直接从github上面下载最新的torchvision，https://github.com/pytorch/vision，点击master，选择tags，注意不要选择Branches！！！！

参考下图，对于torch1.7.0来说最好安装torchvision0.8.1，亲测0.8.0会出现找不到nms的问题。

点进去之后，如下图：

下载完后，将文件先解压，传到nx的桌面，然后执行cd命令到下载完的vision文件夹，执行以下命令：

conda activate yolov5_env

进入虚拟环境后执行：

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

或者执行以下命令：

/home/nvidia/mambaforge/envs/yolov5_env/bin/python setup.py install

耐心等待片刻即可完成编译。

一起开启新世界的大门吧😊