jetson nano安装pytorch
时间: 2023-04-29 12:04:47 浏览: 227
torch1.2.0+torchvision0.4.0.rar
可以使用pip安装PyTorch,在终端中输入以下命令:
```
pip3 install torch torchvision
```
如果你需要使用 CUDA 来加速运算,可以在命令后面添加 `torch.backends.cudnn.enabled=True`:
```
pip3 install torch torchvision torch.backends.cudnn.enabled=True
```
在某些情况下,你需要在命令前面加上 `sudo` 来获得足够的权限进行安装。
如果你已经安装了一个旧版本的 PyTorch,你可能需要使用 `pip3 install torch torchvision --upgrade` 来更新它。
阅读全文