Python get_cuda_cudnn_nccl_choices Beispiele

Beispiel #1

#!/usr/bin/env python

import argparse
import os

import argconfig
import docker
import shuffle

params = {
    'base': docker.base_choices,
    'cuda_cudnn_nccl': docker.get_cuda_cudnn_nccl_choices('cupy',
                                                          with_dummy=True),
    'numpy': ['1.9', '1.10', '1.11', '1.12'],
    'cython': [None, '0.28.0', '0.28.3'],
    'pip': [None, '7', '8', '9', '10'],
    'wheel': [False, True],
}

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()

Beispiel #2

Datei: run_cupy_combination_test.py Projekt: Crissman/chainer-test

#!/usr/bin/env python

import argparse
import os
import random

import argconfig
import docker
import shuffle
import version

params = {
    'base': docker.base_choices,
    'cuda_cudnn_nccl': docker.get_cuda_cudnn_nccl_choices('cupy'),
    'numpy': docker.get_numpy_choices(),
    'scipy': [None, '0.19', '1.0'],
}

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()

Beispiel #3

#!/usr/bin/env python

import argparse
import os

import argconfig
import docker
import shuffle

params = {
    'base':
    docker.base_choices,
    'cuda_cudnn_nccl':
    docker.get_cuda_cudnn_nccl_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.26.1', '0.28.3'],
    '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)

Beispiel #4

Datei: run_combination_test.py Projekt: bupt-renpei/chainer-test

#!/usr/bin/env python2

import argparse

import argconfig
import docker
import shuffle
import version


params = {
    'base': docker.base_choices,
    'cuda_cudnn_nccl': docker.get_cuda_cudnn_nccl_choices('chainer'),
    'numpy': docker.get_numpy_choices(),
    'scipy': [None, '0.18', '0.19', '1.0'],
    'protobuf': ['2', '3', 'cpp-3'],
    'h5py': [None, '2.5', '2.6', '2.7'],
    'pillow': [None, '3.4', '4.0', '4.1'],
    'theano': [None, '0.8', '0.9'],
    'ideep': [
        None,
        version.get_ideep_version_from_chainer_docs()[:3],  # 'major.minor'
    ],
}


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')