Cupy添加在同一台计算机上的不同环境中的不同工作方式

如何解决Cupy添加在同一台计算机上的不同环境中的不同工作方式

我开始实现DL代码，这是我第一次使用DL，GPU和Cupy。我做了一个玩具示例，该示例在同一台机器上的不同python env中的工作方式有所不同，我希望详细了解这种情况的发生原因和发生原因。

这是我的代码

import numpy as np
import cupy as cp

X = 150000
Y = 100
Z = 79

shape_pca_mean = np.random.rand(X,1)

shape_pca_basis = np.random.rand(X,Y)
shape_pca_basis_gpu = cp.asarray(shape_pca_basis)

shape_params = np.random.rand(Y,1)
shape_params_gpu = cp.asarray(shape_params)

shape_x = shape_pca_basis_gpu.dot(shape_params_gpu)

exp_pca_basis = np.random.rand(X,Z)
exp_pca_basis_gpu = cp.asarray(exp_pca_basis)

exp_params = np.random.rand(Z,1)
exp_params_gpu = cp.asarray(exp_params)

exp_x = exp_pca_basis_gpu.dot(exp_params_gpu)

a = shape_pca_mean + cp.asnumpy(exp_x) + cp.asnumpy(shape_x)

我有2个Python环境。全局python3和本地virtualenv python3。它们都具有相同版本的NumPy，cupy，cupy-cuda110和其他安装的python库。上面的代码在两个环境中都能很好地工作。

但是，如果我在下面再添加一行，则virtualenv python3效果很好，但是如果在全局Python上运行，则会出现以下错误。

b = cp.asarray(shape_pca_mean) + exp_x + shape_x

错误

$ python3 scratch.py
Traceback (most recent call last):
  File "scratch.py",line 18,in <module>
    a = cp.asarray(shape_pca_mean) + exp_x + shape_x
  File "cupy/core/core.pyx",line 968,in cupy.core.core.ndarray.__add__
  File "cupy/core/_kernel.pyx",line 951,in cupy.core._kernel.ufunc.__call__
  File "cupy/core/_kernel.pyx",line 974,in cupy.core._kernel.ufunc._get_ufunc_kernel
  File "cupy/core/_kernel.pyx",line 714,in cupy.core._kernel._get_ufunc_kernel
  File "cupy/core/_kernel.pyx",line 61,in cupy.core._kernel._get_simple_elementwise_kernel
  File "cupy/core/carray.pxi",line 196,in cupy.core.core.compile_with_cache
  File "/home/ppinazuredlvm1/.local/lib/python3.6/site-packages/cupy/cuda/compiler.py",line 298,in compile_with_cache
    extra_source,backend)
  File "/home/ppinazuredlvm1/.local/lib/python3.6/site-packages/cupy/cuda/compiler.py",line 352,in _compile_with_cache_cuda
    ls.add_ptr_data(ptx,'cupy.ptx')
  File "cupy/cuda/function.pyx",line 230,in cupy.cuda.function.LinkState.add_ptr_data
  File "cupy/cuda/function.pyx",line 232,in cupy.cuda.function.LinkState.add_ptr_data
  File "cupy/cuda/driver.pyx",line 198,in cupy.cuda.driver.linkAddData
  File "cupy/cuda/driver.pyx",line 118,in cupy.cuda.driver.check_status
cupy.cuda.driver.CUDADriverError: CUDA_ERROR_INVALID_PTX: a PTX JIT compilation failed

如果有帮助，this cupy issue中将提供有关我的系统和驱动程序等的更多详细信息。

看起来不像通常的驱动程序不兼容问题，好像是这种情况，上面的代码不会在任何环境中运行。

解决方法

您可能同时安装了cupy和cupy-cudaXXX，这将导致各种奇怪的问题。如果是这样，请移开杯子，并坚持使用车轮版本。