numba-pro 无法加载 CUDA 设备

Question

我正在尝试确定 GPU 计算是否会显着加快我的代码速度。我已经通过 conda 包管理器安装了 numba-pro。

但是，该模块似乎无法检测到 CUDA 设备：

 In [1]: import numbapro ; numbapro.check_cude()
Vendor:  Continuum Analytics, Inc.
Package: numbapro
Message: trial mode expires in 30 days
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-1-0448b95a9b9c> in <module>()
----> 1 import numbapro ; numbapro.check_cude()

AttributeError: 'module' object has no attribute 'check_cude'

In [2]: import numbapro ; numbapro.check_cuda()
---------------------------------------------------------------------------
CudaSupportError                          Traceback (most recent call last)
<ipython-input-2-f7c0deebb46f> in <module>()
----> 1 import numbapro ; numbapro.check_cuda()

/home/phys/users/jonathan.kadmon/anaconda/envs/adam/lib/python2.7/site-packages/numbapro/__init__.pyc in check_cuda()

/home/phys/users/jonathan.kadmon/anaconda/envs/adam/lib/python2.7/site-packages/numba/cuda/device_init.pyc in is_available()
     19     This will initialize the driver if it hasn't been initialized.
     20     """
---> 21     return driver.driver.is_available and nvvm.is_available()
     22
     23

/home/phys/users/jonathan.kadmon/anaconda/envs/adam/lib/python2.7/site-packages/numba/cuda/cudadrv/driver.pyc in is_available(self)
    187     def is_available(self):
    188         if not self.is_initialized:
--> 189             self.initialize()
    190         return self.initialization_error is None
    191

/home/phys/users/jonathan.kadmon/anaconda/envs/adam/lib/python2.7/site-packages/numba/cuda/cudadrv/driver.pyc in initialize(self)
    182         except CudaAPIError as e:
    183             self.initialization_error = e
--> 184             raise CudaSupportError("Error at driver init: \n%s:" % e)
    185
    186     @property

CudaSupportError: Error at driver init:
Call to cuInit results in CUDA_ERROR_NO_DEVICE:

一些背景：

anaconda 作为用户安装在 linux 机器上。我没有超级用户。
文件系统是 NAS 的一部分，它包含我机构中的所有主文件夹。
anaconda 最初是通过另一台具有相同架构但可能没有 GPU 的机器安装在 NAS 上的。
在有GPU的电脑上我创建了一个新环境并安装了numbapro，它也安装了所有依赖项，包括cudatoolkit
1. 我正在从远程笔记本电脑通过 ssh 连接到机器（可能是当我使用 ssh 时驱动程序没有初始化
2. 我没有安装 lspci，所以我无法检查 GPU，但是我可以通过以下方式查看设备：
  
  猫/proc/driver/nvidia/version
  
  NVRM 版本：NVIDIA UNIX x86_64 内核模块 355.11 Wed Aug 26 16:35:41 PDT 2015 GCC 版本：gcc 版本 4.8.5 (Gentoo 4.8.5 p1.3, pie-0.6.2)

如有任何帮助，我们将不胜感激。这是一个 ssh 问题吗（我将在几天内无法访问机器本身）？包的构建和 link 有问题吗？

Answer 1

的确是ssh的问题。该问题已通过 IT 添加我的用户名解决，该用户名在每次登录时初始化 GPU。

numba-pro 无法加载 CUDA 设备

numba-pro unable to load CUDA device

python

cuda

anaconda

numba-pro