5-pytorch安装

最新推荐文章于 2021-12-15 10:41:48 发布

白宸羽

最新推荐文章于 2021-12-15 10:41:48 发布

阅读量334

点赞数 1

本文链接：https://blog.csdn.net/weixin_50190297/article/details/115474435

版权

本文档介绍了如何在树莓派上下载并安装PyTorch及其依赖包 torchvision。首先，提供了从百度网盘下载的whl文件链接，然后通过FileZilla将文件传输到树莓派。接着，更新库列表并安装libopenblas-dev，再安装numpy和yaml。最后，使用pip3安装whl文件，并通过Python3测试安装是否成功。

摘要由CSDN通过智能技术生成

5-pytorch安装

作者：白宸羽

下载pytorch的whl包

网盘地址：https://pan.baidu.com/s/1CrHaGjkMooplIeM-lXHqPA
提取码:g16q

torch-1.6.0a0+b31f58d-cp37-cp37m-linux_armv7l.whl
torchvision-0.8.0a0+10d5a55-cp37-cp37m-linux_armv7l.whl

把文件导入树莓派

用FileZilla

ip：192.168.43.138
用户：pi
密码：raspberry
端口：22

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

安装依赖包

安装 libopenblas-dev

先更新库列表

sudo apt-get update

安装libopenblas-dev

sudo apt-get install libopenblas-dev

在这里插入图片描述

安装numpy

pip3 install numpy pyyaml

安装torch

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

在这里插入图片描述

安装torchvision

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

在这里插入图片描述

测试

python3

import torch
 
a=torch.rand(5,3)
print(torch.__version__)

a

结果：

1.6.0a0+b31f58d
tensor([[0.2627, 0.3214, 0.9186],
        [0.2076, 0.6210, 0.5768],
        [0.1663, 0.5311, 0.2942],
        [0.9985, 0.0736, 0.5810],
        [0.1169, 0.5002, 0.1826]])