史上最全jetson使用jetpack4.6.1烧录,配置并在虚拟环境配置pytorch,tensorrt (二)

2.1 安装miniforge

Releases · conda-forge/miniforge · GitHub去下载miniforge-pypy3的linux-aarch64安装包

进入到miniforge的sh文件所在目录，右键打开Terminal，输入以下命令进行安装：

sh 文件名.sh

安装的时候会问你要不要初始化,选不要初始化环境变量

安装完毕后，添加环境变量。否则会出现bash:conda Command not found的错误。

# 编辑环境变量
vim ~/.bashrc
# 增加环境变量, 将<username>换成你的用户名
export PATH=/home/<username>/miniforge-pypy3/bin:$PATH
# 激活环境变量
source ~/.bashrc
# 显示(base)
source activate  # 没有显示（base）时使用

更换下载源

conda config --prepend channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --prepend channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

这里换源还是要换的,当然你一直挂着代理也很好用

输入指令：

conda config --set auto_activate_base false

否则(base)会一直在最前面,当然你喜欢开着也行

要用的时候,给一些常见的指令:

source activate #显示(base)
conda activate 你的环境名 #激活环境
conda create -n 环境 --clone 环境 #建议装好一个库就这样备份一个,bug会很多
conda env remove --name your_env_name #删除环境

然后

conda create -n 环境名 python=3.6

创建虚拟环境,后面的操作在虚拟环境激活后进行

2.2 配置虚拟换境

这里还是建议装torch1.8

虽然根据官网的意思PyTorch for Jetson - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums十这些一直到torch1.10都是可以用的,但是版主亲测

torch1.10.0能装,但torchvision0.11.0在pip源里没有

torch1.9.0有bug,好多人反应过了,这个没啥办法

torch1.8有bug 但最后能装上

所以请下载torch1.8的whl文件,

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython

装一下前置,然后进入你whl文件的目录

pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

安装就好

装好了记得调用python

import torch
torch.__version__
torch.cuda.is_available()

测试一下是不是有问题. 博主没遇到问题,但逛论坛看到有这样的问题:

 有些大哥问了,想装别的python版本,想装其他torch怎么办?

Releases · KumaTea/pytorch-aarch64 (github.com)我本来也有类似的需求,在github找到了个人编译的库,然后我操蛋的发现torch.cuda.is_available()返回false,然后我发现了原因:

 大哥没gpu,所以如果想要自己的py版本和pyt版本,请:

#!/usr/bin/env bash

set -xe

VER="$1"
export PYTORCH_BUILD_VERSION="$VER"
export PYTORCH_BUILD_NUMBER="1"

git clone https://github.com/pytorch/pytorch torch || :

cd torch
git checkout "v$VER"
git checkout --recurse-submodules "v$VER"
git submodule sync
git submodule update --init --recursive

rm build/CMakeCache.txt || :
# export MAX_JOBS=1
export BUILD_TEST=0
export USE_BREAKPAD=0

python3 setup.py build
python3 setup.py install
python3 setup.py bdist_wheel

自己在jetson上编译吧!

然后装torchvision,这个没啥好说的,就是去git下载下来自己编译,但是我看内网的大家编译之后的库都在conda list找不到,这里我给个办法:

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download

cd torchvision

export BUILD_VERSION=0.x.0      # where 0.x.0 is the torchvision version  

python3 setup.py install --user

这样就行了.详细的过程请参考这位大佬: 他遇到的报错我也遇到了.

(44条消息) NVIDIA JETSON XAVIER NX （四）安装Pytorch和torchvision_jetson安装pytorch_小可同学呀的博客-优快云博客

后面还要装opencv

Jetson AGX Xavier python虚拟环境+yolo配置+opencv_疯狂的L字母菌的博客-优快云博客

直接把系统自带的opencv复制到虚拟环境下面就行,但这个opencv不支持cuda,需要用cuda的各位当然可以去装,搜一下有的是教程,这里我提出一点问题:如果换了cuda版本,后面要用摄像头最好买usb的,自己装cuda的opencv在使用gsreamer给csi摄像头做流的时候报错