基于ssh连接的Cuda设备上的NumbaPro

Question

我正在使用Python/NumbaPro在windows系统上使用CUDA complient GPU。我使用Cygwin作为shell，从Cygwin控制台中可以找到我的CUDA设备。我用简单的命令进行测试

    numbapro.check_cuda()

但是，当我通过OpenSSH连接到box时（作为Cygwin设置的一部分），我得到以下错误：

^{pr2}$

怎么解决这个问题？在

Answer 1

主要原因是Windows服务会话0隔离。当您通过在会话0（例如sshd或windows远程桌面）中运行的服务运行任何应用程序时，machines本机显示驱动程序不可用。对于CUDA应用程序，这意味着您在运行时会收到一个no device available错误，因为您用于登录的sshd正在作为服务运行，并且没有可用的CUDA驱动程序。在

以下是一些解决方法：

1. 将sshd作为进程而不是服务运行。在
2. 如果您有兼容的GPU，请使用TCC驱动程序，而不是GPU显示驱动程序。在

第二个问题是，您看到的Python运行时错误来自multiprocessing模块。从this question看来，根本原因可能是没有设置NUMBER_OF_PROCESSORS环境变量。您可以使用该线程中的一种解决方法来解决该问题