Jetson xvaier nx 安装torch1.6.0 torchvision0.7.0

已经安装了python3 pip3 

ubuntu18.04

jetpack4.5

cuda10.2

。。。。

1、安装PyTorch1.6.0

链接: https://pan.baidu.com/s/1c-N5JrWT0UbfratLNmzSiQ

提取码: evnm 


百度网盘下载 torch-1.6.0-cp36-cp36m-linux_aarch64.whl文件,之后在.whl文件路径下运行:

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip install Cython
pip install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl

2、安装TorchVision0.7.0
下载TorchVision0.7.0版本文件夹:

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

在vision-0.7.0路径下运行:(注意下载后源码要全部赋予读写权限)

sudo python setup.py install
 

3.安装torch2rt

apt-get install libopenblas-base libopenmpi-dev

pip3 install packaging -i https://pypi.tuna.tsinghua.edu.cn/simple

git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt  (注意下载后源码要全部赋予读写权限)
sudo  python3 setup.py install

你可能感兴趣的:(计算机视觉,深度学习,pytorch,深度学习)