jetson agx orin 的pytorch、torchvision、tensorrt安装最全教程
pytorch安装
conda创建新环境,python版本最好选择3.8
# 创建虚拟环境,其中pytorch3.8是环境名
conda create -n pytorch3.8 python=3.8
输入y继续,等待环境创建完成
# 打开虚拟环境
conda activate pytorch3.8
前面base变成pytorch3.8就是进入了所创建的虚拟环境
pytorch版本不能随意安装,必须安装英伟达编译的好的库文件,可通过以下命令查看Jetpack版本号:
sudo apt-cache show nvidia-jetpack
我是Jetpack5.0.1,选择1.12.0版本的pytorch,pytorch下载的官网地址
点击下载,然后将文件移动到home
开始安装:
安装 PyTorch 所需的系统软件包:
在终端的虚拟环境中输入下列命令
sudo apt-get -y update
sudo apt-get -y install autoconf bc build-essential g++-8 gcc-8 clang-8 lld-8 gettext-base gfortran-