Я пытаюсь запустить код на моем GPU, и сначала я столкнулся с проблемой наличия работающей CUDA на Ubuntu 18.04. После некоторой боли мне предложили поработать с Docker изображением, где у меня есть (я полагаю) хорошо работающая версия CUDA (я начинаю изучать pytorch, и мне удалось получить torch.cuda.is_available() = True).
После этого я попытался запустить простой код, скопированный здесь , чтобы проверить, смог ли мой графический процессор эффективно выполнить свою работу.
К сожалению, нет ... Я получаю сообщение:
NotImplementedError: bounds checking is not supported for CUDA
, и я действительно не смог понять, как исправить это.
Это то, что я получаю, если я запускаю nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.33.01 Driver Version: 440.33.01 CUDA Version: 10.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce 940M On | 00000000:0A:00.0 Off | N/A |
| N/A 36C P0 N/A / N/A | 242MiB / 2004MiB | 1% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
+-----------------------------------------------------------------------------+
И это то, что я получаю, если я запускаю nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243