python-通过ssh连接在Cuda设备上的NumbaPro

我正在使用Python / NumbaPro在窗口框上使用符合CUDA的GPU.我使用Cygwin作为外壳,并且从cygwin控制台中查找我的CUDA设备没有问题.我用简单的命令测试

    numbapro.check_cuda()

但是,当我通过OpenSSH连接到盒子时(作为Cygwin设置的一部分),出现以下错误：

numba.cuda.cudadrv.error.CudaSupportError: Error at driver init:
Call to cuInit results in CUDA_ERROR_NO_DEVICE:

如何解决这个问题？

解决方法:

主要原因是Windows服务会话0隔离.当您通过在会话0中运行的服务(例如sshd或Windows远程桌面)运行任何应用程序时,计算机的本机显示驱动程序不可用.对于CUDA应用程序,这意味着您在运行时会遇到没有设备可用的错误,因为用于登录的sshd作为服务运行,并且没有可用的CUDA驱动程序.

有一些解决方法：

>将sshd作为进程而不是服务运行.
>如果您具有兼容的GPU,请使用TCC驱动程序而不是GPU显示驱动程序.

在第二个问题上,您看到的Python运行时错误来自多处理模块.从this question开始,似乎根本原因可能是未设置NUMBER_OF_PROCESSORS环境变量.您可以使用该线程中的一种解决方法来解决该问题.

标签：python,cuda,ssh,numba-pro
来源： https://codeday.me/bug/20191013/1905089.html