树莓派5 Ubuntu23.10 部署Pytorch记录

最近搞了块pi5，尝试用它来搞图像分类任务，用来跑训练好的模型。发现网上关于pi5的资料和教程还不是很多，pi5所支持的系统也都是最新的，按着现有的一些教程帖子配置环境，就遇到了教程太老不适用的情况。由于还是习惯Ubuntu的系统，本人在折腾了几天raspberry pi os后最终还是换成了Ubuntu，不过树莓派5目前支持的Ubuntu系统只有Ubuntu23.10，老的装不上。

关于在树莓派5 Ubuntu23.10上pytorch的安装，我主要参考了Qengineering/PyTorch-Raspberry-Pi-64-OS: PyTorch installation wheels for Raspberry Pi 64 OS (github.com)下面是我安装的过程，供参考。

首先是安装依赖项 

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install python3-pip libjpeg-dev libopenblas-dev libopenmpi-dev libomp-dev

然后是创建虚拟环境，先检查下系统python版本和路径

python3 --version
which python3.11

根据路径位置编辑.bashrc文件 

echo "export VIRTUALENVWRAPPER_PYTHON=/usr/bin/python3.11" >> ~/.bashrc
source ~/.bashrc

然后安装虚拟环境软件包

sudo apt-get install python3-virtualenv
sudo apt-get install python3-virtualenvwrapper

 

安装完成后还要再编辑下.bashrc文件，这里我遇到了个问题，可能是系统差异，教程里的virtualenvwrapper.sh这个文件被下载到/usr/local/bin/目录下；而我查找文件夹，发现我的virtualenvwrapper.sh文件被下载到了/usr/share目录下，通过修改路径就不报错了

echo "export WORKON_HOME=$HOME/.virtualenvs" >> ~/.bashrc
echo "source /usr/share/virtualenvwrapper/virtualenvwrapper.sh" >> ~/.bashrc

 然后就可以正常创建虚拟环境了,如果要退出虚拟环境输入deactivate，再次进入则输入“workon 环境名称”就行，进入虚拟环境后，用户名左边会多出括号加环境名，这就代表虚拟环境安装成功了。

mkvirtualenv pytorch

然后再在创建的虚拟环境下依次输入以下内容

pip3 install setuptools numpy Cython
pip3 install requests
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
pip3 install torchaudio --index-url https://download.pytorch.org/whl/cpu

安装好后，测试下 打印torch版本，没报错说明安装成功 

最后，把项目移植到树莓派上，尝试能不能跑通

在pycharm上导入之前创建好的本地虚拟环境，运行成功。