Проблемы с импортом torch.nn ⇐ Python

[Anonymous]

Я пытаюсь использовать свой университетский сервер для увеличения вычислительной мощности, однако у меня возникли проблемы с импортом torch.nn. Я попытался создать новую среду на диске (чтобы избежать ее установки в домашнем каталоге по умолчанию), удалить torch и torchvision и установить их снова, но ничего не помогло.

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

>    ModuleNotFoundError                       Traceback (most recent
> call last) Cell In[5], line 1
> ----> 1 import torch.nn
>
> File
> ~\AppData\Roaming\Python\Python311\site-packages\torch\nn\__init__.py:1
> ----> 1 from .modules import *  # noqa: F403
>       2 from .parameter import (
>       3     Parameter as Parameter,
>       4     UninitializedParameter as UninitializedParameter,
>       5     UninitializedBuffer as UninitializedBuffer,
>       6 )
>       7 from .parallel import DataParallel as DataParallel
>
> File
> ~\AppData\Roaming\Python\Python311\site-packages\torch\nn\modules\__init__.py:1
> ----> 1 from .module import Module
>       2 from .linear import Identity, Linear, Bilinear, LazyLinear
>       3 from .conv import Conv1d, Conv2d, Conv3d, \
>       4     ConvTranspose1d, ConvTranspose2d, ConvTranspose3d, \
>       5     LazyConv1d, LazyConv2d, LazyConv3d, LazyConvTranspose1d, LazyConvTranspose2d, LazyConvTranspose3d
>
> File
> ~\AppData\Roaming\Python\Python311\site-packages\torch\nn\modules\module.py:8
>       5 import weakref
>       7 import torch
> ----> 8 from torch._prims_common import DeviceLikeType
>       9 from ..parameter import Parameter
>      10 import torch.utils.hooks as hooks
>
> ModuleNotFoundError: No module named 'torch._prims_common'

Я попробовал установить фонарик, но не знаю, где что-то пошло не так. Я не могу ни импортировать torch.nn, ни проверить версию torch. Я сомневаюсь, что попытка запустить его на сервере является частью проблемы. Я ожидал, что смогу импортировать torch.nn после успешной установки torch.
Любая помощь/руководство будет высоко оценена! Спасибо.

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