如何在Jetson上正确安装pytorch

如何在Jetson上正确安装pytorch

具体请看 Pytorch 官方github

大米力 发表于 2020-5-29 11:46
具体请看 Pytorch 官方github

对于Jetpack 4.2, 4.3, 4.4
目前pytorch的官方文档网址为：
https://elinux.org/Jetson_Zoo#PyTorch%20(Caffe2)

在如上链接，Nvidia提供了针对不同Jetson版本的预编译wheel文件，方便大家直接下载使用。
您可以直接下载wheel文件，并且通过pip install命令来进行安装。

请避免参考二手信息，或者第三方信息来造成不必要的麻烦。