【NVIDIA Jetson NX】环境配置及pytorch安装

关于CUDA的环境变量导入

jetson nano是原装了CUDA的，但是需要用户导入环境变量（导入相关的路径）才可以使用，只有环境变量导入成功后，方可在命令行使用 nvcc -V
在命令行输入 sudo gedit ~/.bashrc
(类似于文本编辑器，需要连接显示器，如果没有显示器，需要自行百度vim的方法来添加环境变量)
在最后添加这三行 （注意，输入变量时，是usr，不是user，linux小白当时差点没注意这点细节）export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH保存后退出，执行 source ~/.bashrc，使得环境变量生效
在命令行输入 nvcc -V 如果正常输出，说明CUDA路径配置成功

torch使用GPU

import torch
from torchvision import transformsdevice = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(torch.cuda.is_available())
print('device %s' % device)
net.to(device=device)

遇到问题：

在 python3.6.9下环境下安装pytorch1.6后，出现过以下问题：

1、AssertionError: Torch not compiled with CUDA enabled
2、UserWarning: Xavier with CUDA capability sm_72 is not compatible with the current PyTorch installation.The current PyTorch install supports CUDA capabilities sm_53.
3、Illegal instruction (core dumped)

第一个问题的解决方案：安装pytorch1.5.0,地址：点击下载

第二个问题的解决方案：算力与版本不匹配，还是安装pytorch1.5.0,地址：点击下载

第三个问题的解决方案：

sudo vim /etc/profile最后一行添加：export OPENBLAS_CORETYPE=ARMV8保存并且退出生效：source /etc/profile

torch下载完成拷贝到NX后，进行安装，注意文件存放位置和执行命令的位置相一致，输入命令：

sudo pip3 install torch-1.5.0-cp36-cp36m-linux_aarch64.whl

torchvision安装

sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch  v0.6.0 https://github.com/pytorch/vision torchvision

直接pip3下载是没有torchvision0.6以及0.7版本，需要git下载，然后编译安装。
cd进入到目录中，git命令下载时修改了文件名称，为torchvision（码云默认下载为vision，无修改）cd路径读者自行判断。

cd torchvision
sudo python3 setup.py install

安装Jetson-stats管理工具

使用Jetson Xavier NX的时候需要查看cpu 电源，gpu的使用 频率和调度等情况，简单的top 命令已经不能满足需求了，有一款软件 jetson-stats ，它相当于主机端使用nvidia-smi来查看显卡使用情况一样，可以完成NX板卡上述资源消耗的使用情况。
在pip配置好的情况下，直接在终端pip3安装pip3 install jetson-stats
安装好后输入jtop查看NX开发套件的各项资源使用情况，按 q 结束退出

刷机预装组件版本检查

1、驱动版本：head -n 1 /etc/nv_tegra_release
2、内核版本：uname -r
3、操作系统：lsb_release -i -r
4、CUDA版本：nvcc -V
5、cuDNN版本：dpkg -l libcudnn8
6、opencv版本：dpkg -l libopencv
7、Tensorrt版本： dpkg -l tensorrt

换源

添加国内清华源，首先备份原本的source.list文件sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak    #为防止误操作后无法恢复，先备份原文件sources.list
sudo gedit /etc/apt/sources.list 然后删除所有内容，复制下列内容到到sources.list后保存deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main multiverse restricted universe
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main multiverse restricted universe
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main multiverse restricted universe
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main multiverse restricted universe
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main multiverse restricted universe
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main multiverse restricted universe
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main multiverse restricted universe
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main multiverse restricted universe之后打开终端输入sudo apt-get update
完成！
一定要注意处理器是aarch64架构的Ubuntu 18.04.2 LTS系统类型的，要使用与之匹配的源 。

参考文献
官方镜像下载
基本环境配置
Pytorch安装
Pytorch官方下载
Archiconda3

【NVIDIA Jetson NX】环境配置及pytorch安装相关推荐

NVIDIA Jetson TX2 环境配置
前言笔者参加学校的冰壶机器人实验课基于NVIDIA Jetson TX2开发板进行开发.这篇文章主要记录NVIDIA Jetson TX2的环境配置过程和踩坑记录. 一.NVIDIA Jetson ...
jetson nano 环境配置（pytorch+ros+rartographer）
1. ,把系统烧录至SD卡后进入系统中依次完成:语言选择.键盘布局选择.WIFI连接.地区选择.用户设置.内存设置(0或者留空). 2. 进入系统设置完成语言的完整下载以及等待系统完成更新.设置系统休 ...
NVIDIA Jetson NX配置深度学习环境（Pytorch、torchvision）超级详细，torchvision安装避坑总结
NVIDIA Jetson NX配置深度学习环境 1.准备python环境 2.安装Pytorch 3.torchvision安装 4.torchvision安装报错问题解决 4.1前言 4.2问题描 ...
NVIDIA AGX Xavier环境配置
NVIDIA AGX Xavier环境配置简介最近有在Xavier上部署算法的需求,熟悉了一下基础环境配置过程,由于踩了不少坑才最后跑出来一个opencv实时跟踪的demo,故此记录一下. Ana ...
【深度之眼PyTorch框架班第五期】作业打卡01：PyTorch简介及环境配置；PyTorch基础数据结构——张量
文章目录任务名称任务简介详细说明作业 1. 安装anaconda,pycharm, CUDA+CuDNN(可选),虚拟环境,pytorch,并实现hello pytorch查看pytorch的 ...
PyTorch强化学习实战（1）——强化学习环境配置与PyTorch基础
PyTorch强化学习实战(1)--强化学习环境配置与PyTorch基础 0. 前言 1. 搭建 PyTorch 环境 2. OpenAI Gym简介与安装 3. 模拟 Atari 环境 4. 模拟 ...
云服务器 Python 环境配置（Pytorch，Yolo）
云服务器 Python & Pytorch 环境配置(Pytorch,Yolo) 1. 服务器环境初始化 1.1 Linux 初始化 sudo apt update` sudo apt upg ...
ubuntu 环境配置与软件安装
ubuntu 环境配置与软件安装 ubuntu系统安装好后的环境配置 1.切换清华源 2.安装nvidia显卡驱动 3.pycharm下载 4.anaconda下载 5.google chrome下载 ...
data2vec进行语音识别验证和微调时的环境配置——flashlight的安装
data2vec进行语音识别验证和微调时的环境配置--flashlight的安装 flashlight/binding/python 依赖安装 gcc7.5.0 安装 CMAKE 安装 FFTW3环境 ...

【NVIDIA Jetson NX】环境配置及pytorch安装

【NVIDIA Jetson NX】环境配置及pytorch安装相关推荐

最新文章

热门文章

【NVIDIA Jetson NX】 环境配置及pytorch安装

【NVIDIA Jetson NX】 环境配置及pytorch安装相关推荐

最新文章

热门文章

【NVIDIA Jetson NX】环境配置及pytorch安装

【NVIDIA Jetson NX】环境配置及pytorch安装相关推荐