Jetson tx1(二) 安装pytorch && torchvision

22次阅读

安装pytorch

确定安装版本

因为没有CUDA版本限制,本博客默认安装最新版(目前为pytorch 1.10)

下载

下载地址,下载需要的版本,推荐在你自己的电脑上下载完之后通过sftp传输到板子上,推荐使用Filezila,这个软件挺好用的。

前置条件准备
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev  
pip3 install Cython 
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

如果没有pip3,执行

sudo apt-get install python3-pip
安装

通过cd切换到whl文件所在位置,执行

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

注意numpy后面的torch要换成你下载的文件,输入torch按下tab键自动补全

验证
import torch
print(torch.__version__)
print('CUDA available: ' + str(torch.cuda.is_available()))
print('cuDNN version: ' + str(torch.backends.cudnn.version()))

安装torchvision

torchvision是一个很有用的工具包,在处理图片等数据时很方便,建议安装

前置条件准备

根据你安装的pytorch版本确定需要安装的torchvision版本,本博客安装的是1.10,故选择torchvision v0.11.1

  • PyTorch v1.0 - torchvision v0.2.2
  • PyTorch v1.1 - torchvision v0.3.0
  • PyTorch v1.2 - torchvision v0.4.0
  • PyTorch v1.3 - torchvision v0.4.2
  • PyTorch v1.4 - torchvision v0.5.0
  • PyTorch v1.5 - torchvision v0.6.0
  • PyTorch v1.6 - torchvision v0.7.0
  • PyTorch v1.7 - torchvision v0.8.1
  • PyTorch v1.8 - torchvision v0.9.0
  • PyTorch v1.9 - torchvision v0.10.0
  • PyTorch v1.10 - torchvision v0.11.1
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.x  # where 0.x.0 is the torchvision version 

以上<version>替换成torchvision版本,例如安装0.11.1

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch 0.11.1 https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
cd torchvision
export BUILD_VERSION=0.11.1  # where 0.x.x is the torchvision version 

实际上是从源码编译安装的过程,git要拉取github上的源码数据,可能下载速度较慢或者不能下载,如果没有科学上网工具的话建议在上午尝试

编译安装

注意:如果是采用python3的话以下三条指令只需要执行第一条即可,

python3 setup.py install --user
cd ../  # attempting to load torchvision from build dir will result in import error
pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

第一条指令需要的时间久一点,因为要编译安装,中间出现的错误不用管,最后显示含successfully installed即安装成功

验证
import torchvision
print(torchvision.__version__)

如果提示no model named PIL,执行如下指令

pip3 install pillow

即可

其他

pip list指令可以列出当前已经安装的模块

这里没有安装tensorboard,因为博主将训练模型的任务交给了电脑,tx1只负责加载模型,运行即可,没必要强求tx1能来训练模型

liubobo
版权声明:本站原创文章,由 liubobo2022-02-07发表,共计2175字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。