没有模块找到火炬

Question

我使用的是带有cuda 10.2.89的jetson NX xavier套件，打开Cv 4.1.1和tensorRT 7.1.3。正在尝试安装pytorch。尝试使用此线路安装

conda install pytorch torchvision cpuonly -c pytorch

但是当我写这行时

import torch

它指出模块未安装，从而引发错误。

如何验证pytorch是否已正确安装。

Answer 1

试试这个

conda install -c pytorch pytorch

执行此命令后，需要输入yes(如果命令行中出现提示(来安装所有相关的软件包。如果在安装库时没有冲突，则将安装PyTorch库。要检查它是否正确安装，请在命令行中键入命令python，然后键入import torch检查它是否安装正确。