Я установил виртуальную среду факела и загрузил сам факел с помощью следующих команд:
(torchgpu) $ pip install --upgrade pip setuptools wheel
(torchgpu) $ pip install --upgrade opencv-python opencv-contrib-python
(torchgpu) $ pip install --upgrade torch torchvision torchaudio
Когда я пытаюсь убедиться, что фонарик установлен правильно, я получаю следующее:
(torchgpu) $ python
>>> import torch
Traceback (most recent call last):
File "", line 1, in
File "/home/USER/venvs/torchgpu/lib/python3.10/site-packages/torch/__init__.py", line 239, in
from torch._C import * # noqa: F403
ImportError: /home/USER/venvs/torchgpu/lib/python3.10/site-packages/torch/lib/../../nvidia/cusparse/lib/libcusparse.so.12: undefined symbol: __nvJitLinkAddData_12_1, version libnvJitLink.so.12
Подробнее здесь: https://stackoverflow.com/questions/786 ... h-c-import
