В Jupyter Notebook нет модуля с именем «torch._custom_ops». ⇐ Python

[Anonymous]

Я не уверен, что это подходящее место для вопроса, но я только что установил инструменты cuda для запуска некоторых функций машинного обучения на основе графического процессора на своем компьютере, и у меня возникла проблема с импортом torch.
Я использую Ubuntu 22.04
Я пробовал установить torch в среде conda и локально. К сожалению, когда я пытаюсь импортировать факел в блокнот Jupyter, я получаю сообщение об ошибке (в блокноте больше ничего не делаю, кроме импорта факела):

Код: Выделить всё

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.
На основании таблицы 3 https://docs.nvidia.com/deploy/cuda-com ... index.html, CUDA 12.4 кажется подходящей версией для моего драйвера NVIDIA.
Я могу запустить python3 -c 'import torch' без вывода, что, как я считаю, хорошо news.
При этом, когда я пытаюсь импортировать факел в блокнот Jupyter, я получаю сообщение об ошибке:

Код: Выделить всё

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

Мне удалось найти torch._custom_ops самому, поэтому я знаю, что он существует, но я не уверен, почему он не работает в Jupyter Notebook?
Я нашел это: загрузка предварительно обученной модели из torch.hub в Sagemaker, но это не казалось актуальным, поскольку я не использую Sagemaker и просто пытаюсь подготовить свою локальную машину к решению задач обучения графического процессора.
/>Я вижу другие сообщения, такие как «Нет модуля с именем «Факел»»

Код: Выделить всё

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

, чего я не делал, но похоже, что это для Windows. Мне это нужно?
Я буду признателен за любую помощь, понимание или просто комментарии, подсказывающие мне, где лучше задать этот вопрос.
Спасибо

Подробнее здесь: https://stackoverflow.com/questions/784 ... -notebooks