Во время работы программы в Ubuntu 18.04 (работает внутри VMWare) я получаю эту ошибку: AssertionError: Факел не скомпилирован с включенной CUDA - PullRequest
Новая
       56

Во время работы программы в Ubuntu 18.04 (работает внутри VMWare) я получаю эту ошибку: AssertionError: Факел не скомпилирован с включенной CUDA

0 голосов
/ 04 февраля 2019

Несмотря на то, что я успешно установил torchvision и cuda, я получаю эту ошибку: Я запускаю программу внутри VMWare (Ubuntu 18.04). 

python3 ./decode/train_dataloader.py --img_data ./datasets/image_fmri --output ./tmp/feat_data
Traceback (most recent call last):
  File "./decode/train_dataloader.py", line 26, in <module>
    net = torchvision.models.vgg19_bn(pretrained=True).cuda()
  File "/home/home/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 260, in cuda
    return self._apply(lambda t: t.cuda(device))
  File "/home/home/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply
    module._apply(fn)
  File "/home/home/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply
    module._apply(fn)
  File "/home/home/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 193, in _apply
    param.data = fn(param.data)
  File "/home/home/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 260, in <lambda>
    return self._apply(lambda t: t.cuda(device))
  File "/home/home/.local/lib/python3.6/site-packages/torch/cuda/__init__.py", line 161, in _lazy_init
    _check_driver()
  File "/home/home/.local/lib/python3.6/site-packages/torch/cuda/__init__.py", line 75, in _check_driver
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

1 Ответ

0 голосов
/ 08 февраля 2019

Проблема решена : ошибка возникла из-за того, что программный код был запущен внутри виртуальной машины (VMWare; ОС: Ubuntu 18.04; Native System OS: Windows 10), в которой отсутствовал драйвер для графики(Поскольку невозможно установить графический драйвер внутри виртуальной машины).

Эта страница является хорошим руководством по установке cuda в Ubuntu: https://www.pugetsystems.com/labs/hpc/How-to-install-CUDA-9-2-on-Ubuntu-18-04-1184/

...