【Jetson系列Orin配置YOLO环境方法】

1.根据官方提示步骤:

        官方教程:Getting Started with Jetson AGX Orin Developer Kit | NVIDIA Developer

        开机->Ctrl+Alt+T打开控制台:

        升级当前设备:

sudo apt dist-upgrade

        完成后重启设备:

sudo reboot

        安装Jetpack:

sudo apt install nvidia-jetpack

        安装完成后:

nvcc -V

        如果出现下图内容则配置成功:

         如果没有的话请自行csdn查找cuda配置环境变量的方法,直至出现上述内容

2.下载安装Archiconda(ARM版Anaconda)

        github地址:Releases · Archiconda/build-tools · GitHub

        进入下载目录后:

sh Archiconda3-0.2.3-Linux-aarch64.sh

        一路按回车直至出现下述内容:

        安装目录会有些问题,如果安装在当前文件夹下的话:

        此处输入--help就会安装到当前文件夹下

        最后再输入一个yes将conda配置进环境变量中就行了

 3.conda换源:

        控制台进入conda安装目录下:

sudo gedit ~/.condarc

         将新的源复制替换保存即可:

        channels:
          - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
          - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
          - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
          - defaults
        show_channel_urls: true

4.进入conda

        控制台输入:

conda

        出现下图则为安装成功:

【Jetson系列Orin配置YOLO环境方法】_第1张图片

         如果没有出现的话请自行csdn配置conda环境变量配置方法

        创建新的虚拟环境:

conda create -n 你自己创建的环境名称 python==3.8.0

        等下载完事就可以了

conda env list

        看一眼能不能看到自己刚创建的虚拟环境就可以

        激活虚拟环境:

conda activate 你自己写的虚拟环境名称

  能看到控制台前面多了一个括号就证明成功进入虚拟环境了

5.环境配置:

        外网教学文档:Jetson Zoo - eLinux.org

        可自行参阅也可以看我的弄

        YOLO无非就是配置tensorflow、opencv、pytorch这三个主要的库,其他库自行pip自动安装就可以了。

         查看Jetpack版本:

apt-cache show nvidia-jetpack

【Jetson系列Orin配置YOLO环境方法】_第2张图片

         可以看到安装的Jetpack版本为5.0.2

        配置设备环境:

sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran

       安装其它库

pip install -U numpy==1.19.4 future==0.18.2 mock==3.0.5 h5py==2.10.0 keras_preprocessing==1.1.1 keras_applications==1.0.8 gast==0.2.2 futures protobuf pybind11

         下载tensorflow2.0(然后pip安装就行了):https://developer.download.nvidia.com/compute/redist/jp/v50/tensorflow/tensorflow-2.8.0+nv22.4-cp38-cp38-linux_aarch64.whl

pip install 把你下载的tensorflow文件拖进来然后按回车就完事了

        下载pytorch1.12.0(安装方法同上):

https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl

        在安装pytorch完成后还要安装torchvision版本号要对应起来:

        torch版本:1.12.0对应torchision版本:0.13.0

        torchision下载:

git clone --branch v0.13.0 https://github.com/pytorch/vision torchvision

        配置安装pytorch的系统环境:

sudo apt-get install libopenblas-base libopenmpi-dev

        还的安装一下Cython库:

pip install Cython

控制台进入下载文件的目录中运行安装(等着就行,时间不短):

export BUILD_VERSION=0.13.0    
python setup.py install --user

        安装完事之后就只剩下安装opencv了:

        参考:安装OpenCV4.6.0简洁过程--Jetson AGX Orin(方便以后安装)_beautifulback的博客-CSDN博客

        最后请自行安装pycharm,运行如下代码:

        

import torch

class check_all:
    """
    检测当前设备torch、tensorflow安装状态
    """
    def __init__(self):
        """
        常量
        """
        torchvision = __import__("torchvision")
        tf          = __import__("tensorflow")
        self.tensorflow_version = tf.__version__
        self.tensorflow_is_gpu  = tf.test.is_built_with_cuda
        self.torch_version = torch.__version__
        self.cuda_version = torch.version.cuda
        self.torchvision_version = torchvision.__version__
        self.use_device = torch.cuda.current_device()
        p = torch.cuda.get_device_properties(self.use_device)
        self.gpu_size = int(p.total_memory/ 1024 ** 3)
        self.device_number = torch.cuda.device_count()
        self.gpu_name = torch.cuda.get_device_name(self.use_device)
        self.gpu_capability = torch.cuda.get_device_capability(self.use_device)

    def check(self):
        """
        检测输出
        """
        print("tensorflow版本{}".format(self.tensorflow_version))
        print("tensorflow是否为CUDA显卡驱动:{}".format("是"if self.tensorflow_is_gpu else "不是"))
        print("CUDA版本:{}".format(self.cuda_version))
        print("pytorch版本:{}\ntorchvision版本:{}".format(self.torch_version,self.torchvision_version))
        print("pytorch是否为CUDA显卡驱动:{}".format("是" if torch.cuda.is_available() else "不是"))
        print("当前设备显卡数量为:{},使用设备编号为:{}".format(self.device_number,self.use_device))
        print("当前使用的设备显卡名称为:{}".format(self.gpu_name))
        print("当前显卡算力为:{}\n当前显卡的内存大小为:{}GB".format(self.gpu_capability,self.gpu_size))



if __name__ == '__main__':
    check = check_all()
    check.check()

输出:

【Jetson系列Orin配置YOLO环境方法】_第3张图片

 安装就完事了,如有问题请留言。

你可能感兴趣的:(python,ubuntu)