Skip to content
Snippets Groups Projects

Py torch

Merged Aurel Istrate requested to merge PyTorch into master
Compare and
6 files
+ 335
1
Preferences
Compare changes
Files
6
##
# This file is an EasyBuild reciPY as per https://github.com/easybuilders/easybuild
#
# Author: Stephane Thiell <sthiell@stanford.edu>
##
easyblock = 'Tarball'
name = 'cuDNN'
version = '7.4.2.24'
cuda_version = '10.0.130'
cuda_version_major_minor = '.'.join(cuda_version.split('.')[:2])
versionsuffix = '-CUDA-%s' % cuda_version
homepage = 'https://developer.nvidia.com/cudnn'
description = """The NVIDIA CUDA Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for
deep neural networks."""
toolchain = {'name': 'dummy', 'version': 'dummy'}
# Nvidia developer registration required.
# Download link: https://developer.nvidia.com/rdp/cudnn-download
# Download as cudnn-10.0-linux-x64-v7.4.2.24.tar
sources = ['%%(namelower)s-%s-linux-x64-v%%(version)s.tar' % cuda_version_major_minor]
dependencies = [('CUDA', cuda_version)]
sanity_check_paths = {
'files': ['include/cudnn.h', 'lib64/libcudnn_static.a'],
'dirs': ['include', 'lib64'],
}
moduleclass = 'numlib'