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

Question

Несмотря на то, что я успешно установил 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

Answer 1

Проблема решена: ошибка возникла из-за того, что программный код был запущен на виртуальной машине (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/