使用 PyCUDA 后无输出

Question

我已经PyCUDA使用pip. 我在两台电脑上试过这个。

一个是全新安装的Python 3.7.1，一个是Python 3.6.5.

使用后一切都失败了PuCUDA，没有错误消息。

最小的例子是这样的：

import sys

import pycuda.driver as cuda

import pycuda.autoinit # <-- Comment in order for `print` to work

if __name__ == '__main__':

    print('Print works')

    sys.stdout.write("Sys print works")

除非我删除pycuda.autoinit.

另一个例子是使用printf：

import pycuda.driver as cuda

import pycuda.autoinit

from pycuda.compiler import SourceModule

if __name__ == '__main__':

    mod = SourceModule("""

        #include <stdio.h>

        __global__ void test() {

          printf("I am %d.%d\\n", threadIdx.x, threadIdx.y);

        }

        """)

    func = mod.get_function("test")

    func(block=(4, 4, 1))

这也不会返回任何输出。

我认为 CUDA 失败了，但没有任何报告。

我的配置：

+--------------------+--------------------+

|        PC1         |        PC2         |

+--------------------+--------------------+

| Python 3.6.5       | Python 3.7.1       |

| Windows 10         | Windows 10         |

| Cuda toolkit 9     | Cuda toolkit 10    |

| GeForce GTX 1050   | GeForce GTX 1080   |

| Visual Studio 2015 | Visual Studio 2015 |

+--------------------+--------------------+

驱动程序：

GeForce Game Ready Driver

Version        : 418.91 WHQL

Release Date   : Wed Feb 13, 2019

我注意到这是一个常见问题，但没有解决方案。

Answer 1

我不确定这是否会对您有所帮助，但通过重新安装cuda（使用anoconda）解决了类似的问题。如果你需要它，你可以通过conda控制台安装它conda install -c anaconda cudatoolkit并检查它是否运行良好，类型&nbsp;numba -s希望这可以帮助