Jetson AGX ORIN 配置 FGVC-PIM 神经网络

sugardisk2024-03-22 13:10

Jetson AGX ORIN 配置 FGVC-PIM 神经网络

文章目录

[Jetson AGX ORIN 配置 FGVC-PIM 神经网络](#Jetson AGX ORIN 配置 FGVC-PIM 神经网络)
- [配置 ORIN 环境](#配置 ORIN 环境)
- [创建 FGVC-PIM 虚拟环境](#创建 FGVC-PIM 虚拟环境)

配置 ORIN 环境

首先先配置 ORIN 的环境，可以参考这个链接：

Jetson AGX ORIN 初始化&配置CUDA&Anaconda&带CUDA的OpenCV

创建 FGVC-PIM 虚拟环境

终端输入命令：

bash 复制代码

conda create -n fgvc-pim python=3.9
conda activate fgvc-pim

即可创建名为 fgvc-pim 的虚拟环境，之后需要向环境中添加需要的安装包

进入 pycharm ，并在 Setting 设置好代码的虚拟环境之后，根据缺少的库的内容，使用 pip install 命令下载库。

需要的库及其命令如下，不需要找了，可以直接下载。

bash 复制代码

pip install numpy pandas matplotlib wandb psutil
# 这个 ORIN 上面是 CUDA/11.4 因此下载的 pytorch 命令是满足 11.4 的版本要求的命令
# 高版本的 PyTorch 一般能兼容低版本的CUDA
pip install torch torchvision torchaudio
pip install opencv-python
pip install scipy scikit-learn

如果下载中断，可以使用镜像源：

bash 复制代码

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pandas

或者使用参数 --default-timeout：

bash 复制代码

pip install --default-timeout=600 pandas

然后再运行训练数据集的命令，发现可以训练了（这个不打算训练，所以 Ctrl + c 强行中断了），配置结束！