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 强行中断了),配置结束!