Jetson AGX ORIN 配置 FGVC-PIM 神经网络
文章目录
- Jetson AGX ORIN 配置 FGVC-PIM 神经网络
- 配置 ORIN 环境
- 创建 FGVC-PIM 虚拟环境
- 安装 PyTorch
- 安装 torchvision
- 安装其他依赖包
配置 ORIN 环境
首先先配置 ORIN 的环境，可以参考这个链接：
Jetson AGX ORIN 初始化&配置CUDA&Anaconda&带CUDA的OpenCV
创建 FGVC-PIM 虚拟环境
终端输入命令：
conda create -n fgvc python=3.8
conda activate fgvc
即可创建名为 fgvc 的虚拟环境，之后需要向环境中添加需要的安装包。python 一定要安装 38 版本，因为安装 PyTorch 的时候需要对照版本进行安装。如果直接使用这个命令进行安装 pip install torch torchvision torchaudio，则会出现无法调用 CUDA 的问题。
安装 PyTorch
安装 PyTorch 之前先要查看一下 ORIN 的版本信息：
- L4T：35.3.1
- Jatpack：5.1.1
然后在官网里面下载安装包。网址如下：
PyTorch for Jetson
我的 ORIN 信息如上，选择对应的 python=3.8 系列，最终确定 PyTorch 版本为 1.12.0，进行下载。
如图所示：
然后激活对应的虚拟环境，进行包的安装。
conda activate fgvc
pip install torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl
显示成功安装后，进行 python 界面，测试是否可以导入，以及 CUDA 是否可用。出现如下结果说明一切正常，可以进行下一步 torchvision 安装。如下所示：
(fgvc) abc@ubuntu:~$ python
Python 3.8.19 (default, Mar 20 2024, 19:53:40)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.12.0a0+2c916ef.nv22.3'
>>> torch.cuda.is_available()
True
>>>
中间可能会出现一个小问题，显示没有 libopenblas-dev 东西，用如下命令进行安装即可解决：
sudo apt-get install libopenblas-dev
安装 torchvision
安装之前先运行以下命令：
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
在环境变量里面添加内容：
export CUDA_HOME=/usr/local/cuda-11.4
# 然后更新
source ~/.bashrc
再看这个对照表
使用 git clone 下载 torchvision 的对应安装代码。
git clone --branch v0.13.0 https://github.com/pytorch/vision torchvision
然后在虚拟环境终端里面输入：
cd torchvision
export BUILD_VERSION=v0.13.0
然后再输入：
python3 setup.py install --user
之后进行调试，还是会有问题，但是具体基本都是一些库需要下载，进行下载即可。
可以参考这个链接：
jetson agx orin 的pytorch、torchvision、tensorrt安装最全教程
# 下载 torchvision 时候报错 需要 numpy
Downloading https://files.pythonhosted.org/packages/51/fe/e4dab289c176ea4e13f97f11f281cc22d4a3b0add9883406db62d4f94d65/numpy-2.0.0b1.tar.gz#sha256=e0bb33a37d0d0b9a19cd41a093877f830e06bd4d989341b9792896cf08e629f7
Best match: numpy 2.0.0b1
Processing numpy-2.0.0b1.tar.gz
error: Couldn't find a setup script in /tmp/easy_install-di_cher0/numpy-2.0.0b1.tar.gz
# 下载 numpy 和 numpy 需要的其他库
(fgvc) abc@ubuntu:~/torchvision$ pip install certifi idna charset-normalizer numpy urllib3
# 参考上面的链接提前安装一些 torchvision 的依赖库
再运行之后问题解除
然后在 python 里面导入 torchvision，看是否有问题，结果如下，能出来 ‘v0.13.0’ 则可以继续安装其他依赖包：
(fgvc) be@ubuntu:~/torchvision$ python
Python 3.8.19 (default, Mar 20 2024, 19:53:40)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
/home/be/torchvision/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")
/home/be/torchvision/torchvision/__init__.py:28: UserWarning: You are importing torchvision within its own root folder (/home/be/torchvision). This is not expected to work and may give errors. Please exit the torchvision project source and relaunch your python interpreter.
warnings.warn(message.format(os.getcwd()))
>>> torch.__version__
'1.12.0a0+2c916ef.nv22.3'
>>> torchvision.__version__
'v0.13.0'
>>>
安装其他依赖包
进入 pycharm，并在 Setting 设置好代码的虚拟环境之后，根据缺少的库的内容，使用 pip install 命令下载库。
需要的库及其命令如下，不需要找了，可以直接下载。
pip install numpy pandas matplotlib wandb psutil
pip install opencv-python
pip install scipy scikit-learn
如果下载中断，可以使用镜像源：
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pandas
或者使用参数 –default-timeout：
pip install --default-timeout=600 pandas
然后再运行训练数据集的命令，发现可以训练了（这个不打算训练，所以 Ctrl + c 强行中断了），配置结束！
