Jupyter Notebooks 中没有名为“torch._custom_ops”的模块

Question

我不确定这是否是正确的提问地点，但我刚刚安装了 cuda 工具来在我的计算机上运行一些基于 GPU 的机器学习内容，并且我遇到了导入 torch 的问题。

我使用的是 Ubuntu 22.04 我尝试在 conda 环境和本地安装 torch。不幸的是，当我尝试将 torch 导入 jupyter 笔记本时，我收到错误（除了导入 torch 之外，在笔记本中实际上什么也没做）：

ModuleNotFoundError: No module named 'torch._custom_ops'; 'torch' is not a package

当我运行

nvcc -V

时，我的输出是：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Mar_28_02:18:24_PDT_2024
Cuda compilation tools, release 12.4, V12.4.131
Build cuda_12.4.r12.4/compiler.34097967_0

当我运行

nvidia-smi

时，我的输出是：

NVIDIA-SMI 550.67                 Driver Version: 550.67         CUDA Version: 12.4 

我有一个 NVIDIA GeForce RTX 3050 Ti。 根据 https://docs.nvidia.com/deploy/cuda-compatibility/index.html 的表 3，CUDA 12.4 似乎是适合我的 NVIDIA 驱动程序的版本。

我能够运行

python3 -c 'import torch'

而没有输出，我认为这是个好消息。 话虽这么说，当我尝试将 torch 导入 jupyter 笔记本时，我收到错误：

ModuleNotFoundError: No module named 'torch._custom_ops'; 'torch' is not a package

我自己找到了 torch._custom_ops，所以我知道它存在，但我不确定为什么它在 Jupyter Notebook 中不起作用？

我发现了这个：从 Sagemaker 中的 torch.hub 加载预训练模型但这似乎并不相关，因为我没有使用 Sagemaker，只是试图让我的本地机器准备好处理 GPU 训练任务。

我看到其他帖子，例如没有名为“Torch”的模块

pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl

，我还没做过，但看起来像是针对Windows的。这是我需要的吗？

如果有任何帮助、见解或简单的评论告诉我一个更好的地方来问这个问题，我将不胜感激。 谢谢你

Answer 1

这只是版本问题...我就是这样

!pip install torch==1.4 torchvision==0.14.1

而且看起来效果很好！