0

我正在尝试确定 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 进入机器（可能在我使用 ssh 时驱动程序未初始化

    2. 我没有安装 lspci，所以我无法检查 GPU，但是我可以通过以下方式查看设备：

      猫 /proc/驱动程序/nvidia/版本

      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 问题吗（我将有几天无法访问机器本身）？这是包的构建和链接的问题吗？

4

1 回答 1

2

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

于 2015-11-15T13:14:35.303 回答