pytorch - 没有找到模块 手电筒

Question

我正在使用具有 cuda 10.2.89、open Cv 4.1.1 和 tensorRT 7.1.3 的 jetson NX xavier 套件。尝试安装 pytorch。尝试使用此行安装

conda install pytorch torchvision cpuonly -c pytorch

但是当我写下这一行时

import torch

它通过说该模块未安装来引发错误。

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

Answer 1

试试这个

conda install -c pytorch pytorch

执行此命令后，您需要输入 yes（如果在命令行中提示）安装所有相关包。如果安装库时没有冲突，将安装 PyTorch 库。要检查它是否正确安装，python请在命令行中键入命令并键入import torch以检查它是否正确安装。