树莓派:树莓派4B+安装Pytorch版本1.6(python3.7)含测试案例

搞了好久,一直失败,拖着过了个年,今天终于安装成功,很开心!“要不断的去尝试,成功是不断尝试出来的。”

正式开始之前可以先看我的这篇文章

https://blog.csdn.net/dujuancao11/article/details/114002979 将树莓派中默认python2切换python3

下载编译好的文件

为了减少下载时间,可直接从我网盘下载。

链接:https://pan.baidu.com/s/1CrHaGjkMooplIeM-lXHqPA 提取码:g16q

利用FileZilla工具实现简单文件传输

链接:https://pan.baidu.com/s/18DAsUVrVh_Lh0ZOUsnlshQ 提取码:2909

将下载好的torch-1.6.0a0+b31f58d-cp37-cp37m-linux_armv7l.whl和torchvision-0.8.0a0+10d5a55-cp37-cp37m-linux_armv7l.whl文件拖到右侧home-->pi


 

安装Pytorch

安装必要的依赖包

sudo apt-get install libopenblas-dev

截屏

 

安装numpy (没有numpy的话也能成功编译,但是编译出来的PyTorch不支持numpy)

pip3 install numpy pyyaml

或者引入源

pip3 install -i https://mirrors.aliyun.com/pypi/simple numpy pyyaml

(我这边是之前已经安装好了)

安装torch

pip install torch-1.6.0a0+b31f58d-cp37-cp37m-linux_armv7l.whl

 

安装torchvision 

pip install torchvision-0.8.0a0+10d5a55-cp37-cp37m-linux_armv7l.whl

 

测试

import torch

a=torch.rand(5,3)
# torch.rand返回服从均匀分布的初始化后的tenosr,外形是其参数size。

print(torch.__version__)

 


版权声明:本文为dujuancao11原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。