jetson 的pytorch 的安装避坑

安装 pytorch,首先 CUDA 的步骤得过一下可以看到 nvcc -V
安装 pytorch 跟 CUDA 的版本要对应
在这里插入图片描述
=网上找了个 1.6.0 的安装包通过 winscp 上传到 jetson 后离线安装下,该离线包可以到资料 5、常用库和模型中获取
sudo pip3 install torch-1.6.0a0+b31f58d-cp36-cp36m-linux_aarch64.whl sudo pip3 install torchvision
sudo pip install boto3

终端输入 python3 进入到 python3 的运行环境中测试下,import torch,我遇到的报错是ImportError:libopenblas.so.0: 无法打 开 共享对 象文件 或 目 录 , 看了下 这 个 教程:https://www.cnpython.com/qa/77454
尝试安装了 OpenBlas 系统库问题解决了

到 python 环境中import torch
print(torch. version )
查看安装的版本

接下来继续在 python3 环境中测试下 pytorch 的功能

输出
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])

正常没有错误哈哈

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

不正经的码狗

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值