NVIDIA AGX Xavier環(huán)境配置

簡(jiǎn)介

最近有在Xavier上部署算法的需求，熟悉了一下基礎(chǔ)環(huán)境配置過(guò)程，由于踩了不少坑才最后跑出來(lái)一個(gè)opencv實(shí)時(shí)跟蹤的demo，故此記錄一下。

Anaconda環(huán)境配置

對(duì)于深度學(xué)習(xí)而言，環(huán)境的隔離還是很重要的，其中Anaconda是本人最常使用的虛擬環(huán)境管理器，由于Xavier是基于ARM架構(gòu)的，常用的Anaconda方法并不適用，這里參考kev的教程，下面簡(jiǎn)述一下安裝過(guò)程。

首先，下載安裝shell腳本，使用下述命令。

wget https://github.com/Archiconda/build-tools/releases/download/0.2.3/Archiconda3-0.2.3-Linux-aarch64.sh

下載之后，使用下述命令進(jìn)行Anaconda的安裝，

bash Archiconda3-0.2.3-Linux-aarch64.sh

在kev的教程中使用sudo權(quán)限進(jìn)行安裝，這樣后續(xù)的環(huán)境創(chuàng)建等有關(guān)的conda命令都需要sudo權(quán)限，所以我這里直接以用戶權(quán)限進(jìn)行安裝。安裝過(guò)程主要是協(xié)議的接受、安裝目錄的選擇（默認(rèn)家目錄）、是否建立conda命令的環(huán)境變量（更新~/.bashrc文件），都同意即可。
此時(shí)輸入conda -V命令查看安裝情況會(huì)找不到命令，需要重開終端或者source ~/.bashrc更新環(huán)境變量。

關(guān)于Anaconda的使用與其他平臺(tái)一致（可以參考我的博客），創(chuàng)建環(huán)境等命令都是相同的。國(guó)內(nèi)源修改使用下述命令即可，前兩條命令是添加兩個(gè)清華源，最后一條命令是設(shè)置安裝包時(shí)源可見(jiàn)。使用國(guó)內(nèi)源會(huì)大大加快環(huán)境創(chuàng)建和包安裝的速度。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes

常用包安裝

關(guān)于包的安裝kev也給出了相關(guān)的教程，需要注意的是，由于Xavier基于ARM架構(gòu)，直接pip安裝的軟件可能下載后無(wú)法安裝，參照kev的教程安裝常用包是比較合理的選擇。

下面重點(diǎn)提兩個(gè)包的安裝，分別是PyTorch和opencv-python。

PyTorch的安裝就需要手動(dòng)下載合適的版本的whl文件，如PyTorch1.4（兼容CUDA10.2）的下載地址如下。具體的不同版本下載地址，這里給出鏈接。

https://nvidia.box.com/shared/static/c3d7vm4gcs9m728j6o5vjay2jdedqb55.whl

下載后使用pip install *.whl命令安裝即可，*代表文件名。我個(gè)人安裝的CUDA10.0所以選擇的torch1.3.0，安裝在虛擬環(huán)境下，安裝后測(cè)試可用性如下圖。

在Xavier不能通過(guò)pip install opencv-python，因?yàn)闆](méi)有相關(guān)的發(fā)行版。需要先手動(dòng)編譯安裝opencv，再在相關(guān)的Python環(huán)境的site-packages目錄下添加文件。

源碼安裝opencv的參考鏈接如下，該鏈接是一個(gè)一鍵安裝的shell腳本，也可以自己按照源碼安裝的思路下載opencv源碼進(jìn)行安裝。

安裝后默認(rèn)會(huì)建立Python2的opencv-python的鏈接，這其實(shí)沒(méi)什么用，一方面Python2已經(jīng)逐漸不被維護(hù)，另一方面即使建立了Python3的鏈接也不會(huì)建立到虛擬環(huán)境的Python環(huán)境中，所以這里需要手動(dòng)操作。

操作代碼如下，其中源文件在系統(tǒng)的Python目錄的dist-packages目錄下（系統(tǒng)Python不同則可能和下述代碼命令不同），目標(biāo)文件夾為當(dāng)前Python環(huán)境的site-packages目錄下。

sudo cp /usr/lib/python3.6/dist-packages/cv2/python-3.6/cv2.cpython-36m-aarch64-linux-gnu.so /home/nvidia/archiconda3/envs/torch/lib/python3.6/site-packages

當(dāng)前虛擬環(huán)境的site-packages目錄通過(guò)激活環(huán)境后which python命令查看，把最后的/bin/python替換為/lib/python3.6/site-packages即可。

使用下述代碼進(jìn)行測(cè)試，結(jié)果如下。

import cv2cap = cv2.VideoCapture(0) while True:ret, frame = cap.read()cv2.imshow('cam', frame)if cv2.waitKey(10) == ord("q"):break cap.release() cv2.destroyAllWindows()

補(bǔ)充說(shuō)明

本文主要介紹Xavier上環(huán)境的基本配置，關(guān)于一些特殊的包的安裝（如scipy）使用pip進(jìn)行編譯安裝時(shí)會(huì)失敗，這里直接使用conda安裝對(duì)應(yīng)包即可。

總結(jié)

以上是生活随笔為你收集整理的NVIDIA AGX Xavier环境配置的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。