安装 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]])
正常没有错误哈哈