0
点赞
收藏
分享

微信扫一扫

jetson nano nvidia GPU ubuntu18 安装pytorch torchvision教程

静悠 2022-02-14 阅读 216

首先切换为国内软件源,然后执行更新操作

sudo apt-get update
sudo apt-get upgrade

安装torch的依赖

sudo apt-get install libopenblas-base libopenmpi-dev

安装  sudo apt-get install python3-pip 

下载官网torch

PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums

Jetson nano自带2.7和3.6版本的python  自带 cuda10.2

我这下载的是 python3支持 的  1.8.0版本 

进入下载.whl的目录,由于pip3 执行下面的命令安装Cpython和使用本地whl包安装torch,注意文件名中的版本要对应。

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

 

打印   Successfully installed torch-1.8.0 

进入python3, 执行代码import torch,如果没有报错则说明torch 即安装成功

安装torchvision

使用git获取torchvision的源码,注意pytorch和torchvision的版本一定要对应

git clone --branch  v0.9.0 https://github.com/pytorch/vision torchvision

编译torchvision

接下来就是编译阶段了,进入torchvision目录,输入命令编译,大致需要花费十分钟。

在编译之前 使用pip/pip3安装 pillow

pip3 install 'pillow<7'

下载速度比较慢时,可以使用国内的豆瓣源下载安装,命令如下所示。

pip3 install -i http://pypi.douban.com/simple --trusted-host pypi.douban.com 'pillow<7'

 

cd torchvision
export BUILD_VERSION=0.x.0  #将x改成自己的版本 
sudo python3 setup.py install --user

 

 

举报

相关推荐

0 条评论