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

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

相关推荐
honeysuckle_luo1 小时前
RandLA-net-pytorch 复现
人工智能·pytorch·python
_BugMan2 小时前
【大模型】理论基础(1):函数与神经网络
人工智能·深度学习·神经网络
hanfeng52682 小时前
使用Deeplabv3+进行遥感影像土地利用分类
深度学习·神经网络·分类
java1234_小锋3 小时前
PyTorch2 Python深度学习 - 自动微分(Autograd)与梯度优化
开发语言·python·深度学习·pytorch2
AI模块工坊3 小时前
CVPR 即插即用 | PConv:重新定义高效卷积,一个让模型“跑”得更快、更省的新范式
人工智能·深度学习·计算机视觉·transformer
java1234_小锋3 小时前
PyTorch2 Python深度学习 - 简介以及入门
python·深度学习·pytorch2
lzjava20244 小时前
Spring AI加DeepSeek实现一个Prompt聊天机器人
人工智能·spring·prompt
fanstuck5 小时前
AI辅助数学建模有哪些优势?
人工智能·数学建模·语言模型·aigc
一只安5 小时前
从零开发AI(不依赖任何模型)
人工智能·python
11年老程序猿在线搬砖6 小时前
如何搭建自己的量化交易平台
大数据·人工智能·python·自动交易·量化交易系统