New package: py-cupy (#19468)
* [py-cupy] created template * [py-cupy] added dependencies * [py-cupy] switched to git based download for submodules * [py-cupy] added dependencies * [py-cupy] added homepage and description. removed fixmes * [py-cupy] updated to tarball from pypi
This commit is contained in:
parent
c2e6610948
commit
0fc79293c6
1 changed files with 27 additions and 0 deletions
27
var/spack/repos/builtin/packages/py-cupy/package.py
Normal file
27
var/spack/repos/builtin/packages/py-cupy/package.py
Normal file
|
@ -0,0 +1,27 @@
|
|||
# Copyright 2013-2020 Lawrence Livermore National Security, LLC and other
|
||||
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
||||
#
|
||||
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
||||
|
||||
from spack import *
|
||||
|
||||
|
||||
class PyCupy(PythonPackage):
|
||||
"""CuPy is an open-source array library accelerated with
|
||||
NVIDIA CUDA. CuPy provides GPU accelerated computing with
|
||||
Python. CuPy uses CUDA-related libraries including cuBLAS,
|
||||
cuDNN, cuRand, cuSolver, cuSPARSE, cuFFT and NCCL to make
|
||||
full use of the GPU architecture."""
|
||||
|
||||
homepage = "https://cupy.dev/"
|
||||
url = "https://pypi.io/packages/source/c/cupy/cupy-8.0.0.tar.gz"
|
||||
|
||||
version('8.0.0', sha256='d1dcba5070dfa754445d010cdc952ff6b646d5f9bdcd7a63e8246e2472c3ddb8')
|
||||
|
||||
depends_on('python@3.6:', type=('build', 'run'))
|
||||
depends_on('py-setuptools', type='build')
|
||||
depends_on('py-fastrlock@0.3:', type=('build', 'run'))
|
||||
depends_on('py-numpy@1.15:', type=('build', 'run'))
|
||||
depends_on('cuda')
|
||||
depends_on('nccl')
|
||||
depends_on('cudnn')
|
Loading…
Reference in a new issue