CuPy 不适用于 Ubuntu 18.04 和 CUDA 9.0

Question

状况：

• CuPy 版本 7.0.0
• 操作系统/平台 Ubuntu 18.04
• CUDA 9.0 版
• cuDNN/NCCL 7.6.5 版（适用于 cuda 9.0）
• GPU 英伟达 GTX580
• 驱动程序版本 390.116

代码 1：

import cupy as cp
x = cp.arange(6).reshape(2, 3).astype('f')

错误信息 1：

NVRTC compilation error: nvrtc: error: invalid value for --gpu-architecture (-arch)

-----
Name:
Options: -I/home/liu/.local/lib/python2.7/site-packages/cupy/core/include -I /usr/local/cuda-9.0/include -ftz=true -arch=compute_20
CUDA source:
1
-----
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/liu/.local/lib/python2.7/site-packages/cupy/creation/ranges.py", line 57, in arange
    _arange_ufunc(typ(start), typ(step), ret, dtype=dtype)
  File "cupy/core/_kernel.pyx", line 864, in cupy.core._kernel.ufunc.__call__
  File "cupy/core/_kernel.pyx", line 876, in cupy.core._kernel.ufunc._get_ufunc_kernel
  File "cupy/core/_kernel.pyx", line 638, in cupy.core._kernel._get_ufunc_kernel
  File "cupy/core/_kernel.pyx", line 60, in cupy.core._kernel._get_simple_elementwise_kernel
  File "cupy/core/carray.pxi", line 166, in cupy.core.core.compile_with_cache
  File "/home/liu/.local/lib/python2.7/site-packages/cupy/cuda/compiler.py", line 136, in compile_with_cache
    base = _preprocess('', options, arch)
  File "/home/liu/.local/lib/python2.7/site-packages/cupy/cuda/compiler.py", line 97, in _preprocess
    result = prog.compile(options)
  File "/home/liu/.local/lib/python2.7/site-packages/cupy/cuda/compiler.py", line 250, in compile
    raise CompileException(log, self.src, self.name, options)
cupy.cuda.compiler.CompileException: nvrtc: error: invalid value for --gpu-architecture (-arch)

代码 2：

import cupy as cp
cp.random.random_sample(10)

错误信息 2：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/liu/.local/lib/python3.6/site-packages/cupy/random/sample.py", line 161, in random_sample
    return rs.random_sample(size=size, dtype=dtype)
  File "/home/liu/.local/lib/python3.6/site-packages/cupy/random/generator.py", line 540, in random_sample
    out = self._random_sample_raw(size, dtype)
  File "/home/liu/.local/lib/python3.6/site-packages/cupy/random/generator.py", line 528, in _random_sample_raw
    func(self._generator, out.data.ptr, out.size)
  File "cupy/cuda/curand.pyx", line 158, in cupy.cuda.curand.generateUniformDouble
  File "cupy/cuda/curand.pyx", line 162, in cupy.cuda.curand.generateUniformDouble
  File "cupy/cuda/curand.pyx", line 86, in cupy.cuda.curand.check_status
cupy.cuda.curand.CURANDError: CURAND_STATUS_LAUNCH_FAILURE

Answer 1

GeForce GTX 580 的计算能力是 2.0，但它不受 CuPy 支持，也不受任何比 CUDA 8.0 更新的 CUDA 工具包版本支持。

https://docs-cupy.chainer.org/en/latest/install.html#requirements

GPU 的计算能力必须至少为 3.0。

https://docs.nvidia.com/cuda/archive/9.0/cuda-toolkit-release-notes/index.html#unsupported-features

费米架构支持。CUDA 工具包，包括 CUDA 编译器 (nvcc)、开发人员工具和 CUDA 库，不再支持 Fermi 架构 (sm_2.x)。请注意，正在从 CUDA 工具包中删除对 Fermi 架构的支持，但未从驱动程序中删除。使用 CUDA 8 或更早版本编译的应用程序将继续使用更新的 NVIDIA 驱动程序在 Fermi 架构上运行。