TensorRT-4.0.1.6-fosscuda-2018b-Python-2.7.15.eb

name = 'TensorRT'
version = '4.0.1.6'
versionsuffix = '-Python-%(pyver)s'

homepage = 'https://developer.nvidia.com/tensorrt'
description = """NVIDIA TensorRT is a platform for high-performance deep learning inference"""

toolchain = {'name': 'fosscuda', 'version': '2018b'}

# Requires manual download from 
# https://developer.nvidia.com/nvidia-tensorrt-4x-download
sources = ['%(name)s-%(version)s.Ubuntu-16.04.4.x86_64-gnu.cuda-9.2.cudnn7.1.tar.gz']
checksums = ['fc5fd2ba86225ed62e60870cc24896107770a59829363fcb49229c82bb1f4fb5']

builddependencies = [
    ('protobuf', '3.6.1'),
]

dependencies = [
    ('Python', '2.7.15'),
    ('cuDNN', '7.1.4.18'),
]

exts_list = [
    ('protobuf-python', '3.6.1', {
        'source_urls': ['https://github.com/google/protobuf/releases/download/v%(version)s/'],
        'source_tmpl': 'protobuf-python-%(version)s.tar.gz',
        'checksums': ['40f009cb0c190816a52fc21d45c26558ee7d63c3bd511b326bd85739b2fd99a6'],
        'use_pip': False,  # installation using pip fails
        'start_dir': 'python',
        'modulename': 'google.protobuf',
    }),
]

moduleclass = 'lib'