jetson nano安装pytorch
时间: 2023-09-30 07:03:24 浏览: 178
pytorch-torchvison.zip
安装 PyTorch 之前,请确保你的 Jetson Nano 已经正确配置了 NVIDIA JetPack SDK。以下是安装 PyTorch 的步骤:
1. 首先,打开终端并进入你的虚拟环境(如果你已经创建了一个)。
2. 安装依赖项:
```bash
sudo apt-get update
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
```
3. 安装 PyTorch:
```bash
pip3 install torch torchvision torchaudio -f https://nvidia.box.com/shared/static/py3h9yq3h4f3vfg0tjzwdarbnfy5a9bo.whl
```
4. 确认 PyTorch 安装成功:
```bash
python3 -c "import torch; print(torch.__version__)"
```
如果输出了 PyTorch 的版本号,则说明安装成功。
现在你已经成功在 Jetson Nano 上安装了 PyTorch。
阅读全文