zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-timm: add new package (#25019)

Browse Source
This commit is contained in:
Adam J. Stewart 2021-07-21 14:25:04 -05:00 committed by GitHub
parent 5a49264e19
commit df8e51d6e5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
var/spack/repos/builtin/packages/py-timm/package.py Normal file
View File

@ -0,0 +1,20 @@
# Copyright 2013-2021 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 PyTimm(PythonPackage):
"""(Unofficial) PyTorch Image Models."""
homepage = "https://github.com/rwightman/pytorch-image-models"
pypi = "timm/timm-0.4.12.tar.gz"
version('0.4.12', sha256='b14be70dbd4528b5ca8657cf5bc2672c7918c3d9ebfbffe80f4785b54e884b1e')
depends_on('python@3.6:', type=('build', 'run'))
depends_on('py-setuptools', type='build')
depends_on('py-torch@1.4:', type=('build', 'run'))
depends_on('py-torchvision', type=('build', 'run'))