Python get_cuda_libs_choices Examples

Example #1

File: run_cupy_combination_test.py Project: asi1024/chainer-test

#!/usr/bin/env python

import argparse
import os
import random

import argconfig
import docker
import shuffle
import version

params = {
    'base': None,
    'cuda_libs': docker.get_cuda_libs_choices('cupy'),
    'numpy': docker.get_numpy_choices(),
    'scipy': docker.get_scipy_choices(),
}

if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        description='Test script for multi-environment')
    parser.add_argument('--id', type=int, required=True)
    parser.add_argument('--no-cache', action='store_true')
    parser.add_argument('--timeout', default='2h')
    parser.add_argument('--interactive', action='store_true')
    parser.add_argument('--clone-chainer',
                        action='store_true',
                        help='clone chainer repository based on cupy version.')

    argconfig.setup_argument_parser(parser)
    args = parser.parse_args()

Example #2

File: run_install_test.py Project: sunxingxingtf/chainer-test

#!/usr/bin/env python

import argparse
import os

import argconfig
import docker
import shuffle
import version

params = {
    'base': None,
    'cuda_libs': docker.get_cuda_libs_choices('chainer', with_dummy=True),
    'numpy': ['1.9', '1.10', '1.11', '1.12'],
    # Chainer does not require Cython, so it should be able to be installed
    # with any Cython version.
    'cython': [None, '0.28.0', '0.29.13'],
    'pip': [None, '7', '8', '9'],
}

if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        description='Test script for installation')
    parser.add_argument('--id', type=int, required=True)

    parser.add_argument('--no-cache', action='store_true')
    parser.add_argument('--timeout', default='2h')
    parser.add_argument('-i', '--interactive', action='store_true')

    argconfig.setup_argument_parser(parser)
    args = parser.parse_args()

Example #3

File: run_combination_test.py Project: sunxingxingtf/chainer-test

#!/usr/bin/env python

import argparse

import argconfig
import docker
import shuffle
import version

ideep_min_version = version.get_ideep_version_from_chainer_docs()
assert ideep_min_version is not None

params = {
    'base': None,
    'cuda_libs': docker.get_cuda_libs_choices('chainer'),
    'numpy': docker.get_numpy_choices(),
    'scipy': [None, '0.18', '0.19', '1.0'],
    'protobuf': ['3', 'cpp-3'],
    'h5py': [None, '2.5', '2.6', '2.7', '2.8', '2.9', '2.10'],
    'pillow': [None, '3.4', '4.0', '4.1', '6.2'],
    'theano': [None, '0.8', '0.9', '1.0'],
    'ideep': [
        None,
        ideep_min_version[:3],  # 'major.minor'
    ],
}

if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        description='Test script for multi-environment')
    parser.add_argument('--id', type=int, required=True)