|
|
发表于 2024-6-17 17:22:59
1376 浏览 1 回复
关于orin nx安装pytorch的问题
[color=rgba(0, 0, 0, 0.85)]想装一下pytorch，根据手册安装的cuda版本是11.4，请问应该安装什么版本的python，以及用什么指令安装pytorch
|
扫一扫浏览分享
|
|
|
|
|
|
|
|
|
|
|
发表于 2024-6-19 09:55:08
本帖最后由 amov_daniel 于 2024-6-19 10:01 编辑
你好
jetson上安装Pytorch，通常和jetpack版本有关，详情请参考链接：https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048。
用户可根据jetpack版本以及其他一些信息，选择对应Pytorch版本并下载安装；然后，torchvision安装的版本需要与pytorch对应，需要源码安装。
另外，我们最新的SpireCV开发者套件已经预装了Pytorch、torchvision等深度学习环境，满足“开箱即用”。
torchvision安装的版本与pytorch对应关系：
|
|
|
|
|
|
|
|
|
|
|
