Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....

简介: 本文提供了在Ubuntu 18.04操作系统的NVIDIA Jetson平台上安装深度学习和计算机视觉相关库的详细步骤,包括PyTorch、OpenCV、ONNX、TensorFlow等。

1.安装pytorch

  • 对于安装pytorch你要结合你的cuda的版本,以及你的python版本,由于我是jetson nano的平台,所以安装的时候要注意有aarch64的后缀,pytorch就是蛮难下下来的,需要你翻墙才能下载,下面我会给出相应的pytorch的腾讯微云分享版本

这是全部的下载链接,直接找对应下载:
https://elinux.org/Jetson_Zoo#PyTorch_.28Caffe2.29

首先给出下载pytorch的代码,如果运行不起就直接https://nvidia.box.com/shared/static/j2dn48btaxosqp0zremqqm8pjelriyvs.whl

wget https://nvidia.box.com/shared/static/j2dn48btaxosqp0zremqqm8pjelriyvs.whl -O torch-1.6.0-cp36-cp36m-linux_aarch64.whl

sudo pip3 install torch-1.6.0-cp36-cp36m-linux_aarch64.whl
  • ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory
  • solve: sudo apt-get install libopenblas-dev
  • OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory
  • solve:
    在这里插入图片描述

安装完了之后就可以通过下面的这个验证

import torch
print(torch.__version__)
print('CUDA available: ' + str(torch.cuda.is_available()))
a = torch.cuda.FloatTensor(2).zero_()
print('Tensor a = ' + str(a))
b = torch.randn(2).cuda()
print('Tensor b = ' + str(b))
c = a + b
print('Tensor c = ' + str(c))

2.安装torchvision

在这里插入图片描述

# 安装一些必要的依赖包
sudo apt-get install libjpeg-dev zlib1g-dev

# 下载torchvision
git clone --branch v0.7.0 https://gitee.com/zero-one-game/vision torchvision   

# 安装torchvision
cd torchvision
sudo python3 setup.py install
# OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory  ->>>>sudo apt-get install libopenmpi2
cd ../
pip3 install 'pillow<7'
  • error: command ‘aarch64-linux-gnu-gcc‘ failed with exit status 1
  • way1: sudo apt-get install python3-dev python3-pip libxml2-dev libxslt1-dev zlib1g-dev libffi-dev libssl-dev
  • way2: if has ffmpeg 改为if False即可

用以下代码测试是否成功

import torchvision

print(torchvision.__version__)

在Jetson Xavier NX上安装torchvision编译报错:fatal error: libavcodec/avcodec.h: No such file or directory
look at the authou csdn

3.安装opencv-python(自带有)

一条命令直接搞定

sudo apt-get install python3-opencv

通过终端验证直接通过

Python 3.6.9 (default, Oct  8 2020, 12:12:24) 
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import cv2
>>>

若要删除用sudo apt-get remove python3-opencv

4.安装onnx/onnxruntime

  • 先安装onnx的环境依赖
    sudo apt-get install protobuf-compiler libprotoc-dev
  • 再安装onnx为1.9.0的版本
    pip install onnx1.9.0
    pip3 install onnxruntime1.9.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/ --trusted-host pypi.tuna.tsinghua.edu.cn
  Created wheel for onnx: filename=onnx-1.4.1-cp36-cp36m-linux_aarch64.whl size=3646291 sha256=3632a794d71e5c744920b2abf7626a4fd3ca39b7ea47a85e7cb9f3a7c325dbf7
  Stored in directory: /home/lqs/.cache/pip/wheels/45/d1/9f/9508c4176d79c0373629eff6cf2645e2dcbfb291bbe9dc2f8a
Successfully built onnx
Installing collected packages: onnx
Successfully installed onnx-1.4.1

5.安装tensorflow-gpu

if your platform is jetson xavier nx,I think you should see the offcial word,the website is :https://forums.developer.nvidia.com/t/official-tensorflow-for-jetson-agx-xavier/65523

jetson nano look at here

1.sudo apt install python3-pip python3-dev
2.python3 -m pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple
#使用下述指令可以更改python镜像源为清华pypi镜像源
3.pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
4.#安装机器学习常用包 
sudo apt install python3-scipy -y
sudo apt install python3-pandas -y
sudo apt install python3-sklearn -y
sudo apt install python3-seaborn -y
sudo pip install cython
上述指令会自动安装numpy和matplotlib等依赖包
5.
sudo apt-get update
sudo apt install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
6.#安装Python依赖包
sudo pip3 install -U numpy==1.16.1 future==0.18.2 mock==3.0.5 h5py==2.10.0 keras_preprocessing==1.1.1 keras_applications==1.0.8 gast==0.2.2 futures protobuf pybind11 astor
7.#download tensorflow https://developer.download.nvidia.com/compute/redist/jp/v45 
pip3 install tensorflow-1.15.5+nv21.4-cp36-cp36m-linux_aarch64.whl --user
8.test
import tensorflow as tf
# 输出提示:
# 2020-10-11 15:25:36.253267: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.10.2

a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
# 输出结果:
# tf.Tensor(3.0, shape=(), dtype=float32)

print('GPU:', tf.test.is_gpu_available())
# 输出最后一句为:
# GPU: True
#this is NV
pip3 install tensorflow-1.15.5+nv21.3-cp36-cp36m-linux_aarch64.whl 
# jetson xavier nx  jetpack 4.5.2 cuda 10.2
  • 验证
    在这里插入图片描述

6.安装setuptools

pip install --no-cache-dir setuptools==20.7.0 -i https://pypi.doubanio.com/simple/
#下面是升级 
pip install --ignore-installed  setuptools

7.安装Pycuda

下载地址
https://files.pythonhosted.org/packages/5e/3f/5658c38579b41866ba21ee1b5020b8225cec86fe717e4b1c5c972de0a33c/pycuda-2019.1.2.tar.gz

tar zxvf pycuda-2019.1.2.tar.gz    
cd pycuda-2019.1.2/  
python3 configure.py --cuda-root=/usr/local/cuda-10.2
sudo python3 setup.py install
  • 若报错FileNotFoundError: [Errno 2] No such file or directory: ‘nvcc’
  • 打开Pycuda的compiler.py文件中的compile_plain()
    中,大约在第 73 行的位置中加入下面段代码!
    在这里插入图片描述

8. 安装Cmake

wget http://www.cmake.org/files/v3.13/cmake-3.13.0.tar.gz
tar xpvf cmake-3.13.0.tar.gz cmake-3.13.0/  
cd cmake-3.13.0/
./bootstrap --system-curl    
make -j4 
echo 'export PATH=/home/z/Downloads/cmake-3.13.0/bin/:$PATH' >> ~/.bashrc
source ~/.bashrc

9. 安装matplotlib

sudo apt-get install python3-matplotlib

10. 安装pyqt5

sudo apt install pyqt5*
sudo apt-get install python3-pyqt5
相关实践学习
部署Stable Diffusion玩转AI绘画(GPU云服务器)
本实验通过在ECS上从零开始部署Stable Diffusion来进行AI绘画创作,开启AIGC盲盒。
目录
相关文章
|
2月前
|
并行计算 PyTorch TensorFlow
Ubuntu安装笔记(一):安装显卡驱动、cuda/cudnn、Anaconda、Pytorch、Tensorflow、Opencv、Visdom、FFMPEG、卸载一些不必要的预装软件
这篇文章是关于如何在Ubuntu操作系统上安装显卡驱动、CUDA、CUDNN、Anaconda、PyTorch、TensorFlow、OpenCV、FFMPEG以及卸载不必要的预装软件的详细指南。
4604 3
|
2月前
|
Ubuntu Linux C语言
Ubuntu安装笔记(二):ubuntu18.04编译安装opencv 3.4.0 opencv_contrib3.4.0
本文介绍了在Ubuntu 18.04系统上编译安装OpenCV 3.4.0及其扩展包opencv_contrib 3.4.0的详细步骤,包括下载源码、安装依赖、配置CMake和编译安装,以及常见问题的解决方法。
180 1
Ubuntu安装笔记(二):ubuntu18.04编译安装opencv 3.4.0 opencv_contrib3.4.0
|
2月前
|
Kubernetes Ubuntu Docker
从0开始搞K8S:使用Ubuntu进行安装(环境安装)
通过上述步骤,你已经在Ubuntu上成功搭建了一个基本的Kubernetes单节点集群。这只是开始,Kubernetes的世界广阔且深邃,接下来你可以尝试部署应用、了解Kubernetes的高级概念如Services、Deployments、Ingress等,以及探索如何利用Helm等工具进行应用管理,逐步提升你的Kubernetes技能树。记住,实践是最好的老师,不断实验与学习,你将逐渐掌握这一强大的容器编排技术。
236 1
|
2月前
|
Python
Jetson环境安装(二):ubuntu18.0卸载和安装python3
在Jetson Nano上如何卸载Python 3.7并重新安装Python 3.7.0版本的详细步骤,包括卸载命令、安装依赖库、下载和编译Python源码以及建立软链接等。
69 2
|
2月前
|
并行计算 PyTorch TensorFlow
环境安装(一):Anaconda3+pytorch1.6.0+cuda10.0+cudnn7.6.4+tensorflow1.15+pycocotools+pydensecrf
这篇文章详细介绍了如何在Anaconda环境下安装和配置深度学习所需的库和工具,包括PyTorch 1.6.0、CUDA 10.0、cuDNN 7.6.4、TensorFlow 1.15、pycocotools和pydensecrf,并提供了pip国内镜像源信息以及Jupyter Notebook和Anaconda的基本操作。
181 0
环境安装(一):Anaconda3+pytorch1.6.0+cuda10.0+cudnn7.6.4+tensorflow1.15+pycocotools+pydensecrf
|
4月前
|
消息中间件 Ubuntu Java
在Ubuntu 18.04上安装Apache Kafka的方法
在Ubuntu 18.04上安装Apache Kafka的方法
202 0
|
4月前
|
存储 Ubuntu Go
在Ubuntu 16.04上安装Go 1.6的方法
在Ubuntu 16.04上安装Go 1.6的方法
51 1
|
4月前
|
存储 Ubuntu Go
在Ubuntu 18.04上安装Go的方法
在Ubuntu 18.04上安装Go的方法
62 1
|
4月前
|
存储 Ubuntu Linux
在Ubuntu 14.04上安装Go 1.6的方法
在Ubuntu 14.04上安装Go 1.6的方法
57 1
|
4月前
|
Ubuntu Linux 持续交付
在Ubuntu 16.04上安装Docker Compose的方法
在Ubuntu 16.04上安装Docker Compose的方法
109 1