numba-pro无法加载CUDA设备

时间:2015-11-13 16:56:46

标签: python cuda anaconda numba-pro

如果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:

一些背景知识:

  1. anaconda作为用户安装在linux机器上。我没有超级用户。
  2. 文件系统是包含我所在机构中所有主文件夹的NAS的一部分。
  3. anaconda最初是通过具有相同架构但可能没有GPU的不同机器安装在NAS上的。
  4. 在带有GPU的计算机上我创建了一个新环境并安装了numbapro,它还安装了所有依赖项,包括cudatoolkit

    1. 我从远程笔记本电脑进入机器(当我使用ssh时可能没有初始化驱动程序
    2. 我没有安装lspci所以我无法检查GPU但是我可以看到设备:

      cat / proc / driver / nvidia / version

      NVRM版本:NVIDIA UNIX x86_64内核模块355.11 Wed 8月26日16:35:41 PDT 2015 GCC版本:gcc版本4.8.5(Gentoo 4.8.5 p1.3,pie-0.6.2)

  5. 任何帮助都将受到高度赞赏。这是一个ssh问题(几天后我将无法访问机器本身)?这是包的构建和链接的问题吗?

1 个答案:

答案 0 :(得分:2)

确实这是ssh的一个问题。 这个问题通过IT添加我的用户名来解决,该用户名在每次登录时初始化GPU。