【发布时间】：2020-01-31 23:31:40
【问题描述】：
我试图在我的 GPU 上运行代码，首先我遇到了在 ubuntu 18.04 上运行 CUDA 的问题。在经历了一些痛苦之后，我被建议在 Docker Image 中工作，在那里我（我认为）有一个运行良好的 CUDA 版本（我开始学习 pytorch，并且我能够获得 torch.cuda.is_available() = True）。
之后我尝试运行一个简单的代码复制here，以查看我的GPU是否能够有效地完成它的工作。
很遗憾没有...我收到消息：
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
【问题讨论】：