Question

我目前正在研究Nvidia Jetson Nano，对Linux不太熟悉。我正在尝试运行一个python文件，该文件会导入一个名为torch的程序包。我已按照NVIDIA here的说明将它与Torchvision一起安装。

当我在终端上运行pip list时，可以看到火炬列为已安装的软件包之一。但是，由于出现以下错误，我无法运行python文件。当我尝试在python shell上运行它时，会弹出相同的错误。仅供参考：以前它有问题，因为系统默认情况下使用的是python 2，但我已经通过编辑.bashrc文件切换到python 3来修复了路径。

>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/jiayi/.local/lib/python3.6/site-packages/torch/__init__.py", line 81, in <module>
    from torch._C import *
ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory

我尝试通过pip卸载和安装，但无济于事。当我尝试安装pytorch软件包时（按照github存储库here中的说明进行操作），发生如下所示的错误，这是由于相同的问题所致。能够检测到割炬软件包已安装，但似乎存在内部问题。

Requirement already satisfied: torch==1.4.0 from file:///home/jiayi/jetson-inference/build/torch-1.4.0-cp36-cp36m-linux_aarch64.whl in /home/jiayi/.local/lib/python3.6/site-packages (1.4.0)
[jetson-inference]  cloning torchvision...
[sudo] password for jiayi: 
Cloning into 'torchvision-36'...
remote: Enumerating objects: 71, done.
remote: Counting objects: 100% (71/71), done.
remote: Compressing objects: 100% (56/56), done.
remote: Total 8219 (delta 37), reused 29 (delta 15), pack-reused 8148
Receiving objects: 100% (8219/8219), 10.22 MiB | 3.60 MiB/s, done.
Resolving deltas: 100% (5631/5631), done.
[jetson-inference]  building torchvision for Python 3.6...
Traceback (most recent call last):
  File "setup.py", line 14, in <module>
    import torch
  File "/home/jiayi/.local/lib/python3.6/site-packages/torch/__init__.py", line 81, in <module>
    from torch._C import *
ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory

[jetson-inference]  installation complete, exiting with status code 0
[jetson-inference]  to run this tool again, use the following commands:

    $ cd <jetson-inference>/build
    $ ./install-pytorch.sh

Answer 1

我遇到了完全相同的问题。问题似乎是cuda 10.2。降级到10.0也无济于事。可能的解决方案是手动安装Jetpack的所有内容，并确保要安装的cuda版本为10.0。

Answer 2

您可以通过检查torch.version.cuda并确保它与您的jetson nano上的cuda版本相同，来检查是否安装了支持cuda的火炬版本正确。

安装割炬的一种更简单的方法是从Jetson Zoo下载.whl文件。将nano升级到最新的Jetpack版本也可能有用

无法导入火炬（ImportError：libcudart.so.10.0）

2 个答案: