import argparse
import os


lowerCAmelCase__ :			str         ={
    '''pytorch''': {
        '''1.13''': {
            '''torch''': '''1.13.1''',
            '''torchvision''': '''0.14.1''',
            '''torchaudio''': '''0.13.1''',
            '''python''': 3.9,
            '''cuda''': '''cu116''',
            '''install''': (
                '''python3 -m pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1'''
                ''' --extra-index-url https://download.pytorch.org/whl/cu116'''
            ),
            '''base_image''': '''nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04''',
        },
        '''1.12''': {
            '''torch''': '''1.12.1''',
            '''torchvision''': '''0.13.1''',
            '''torchaudio''': '''0.12.1''',
            '''python''': 3.9,
            '''cuda''': '''cu113''',
            '''install''': (
                '''python3 -m pip install --no-cache-dir -U torch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1'''
                ''' --extra-index-url https://download.pytorch.org/whl/cu113'''
            ),
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        '''1.11''': {
            '''torch''': '''1.11.0''',
            '''torchvision''': '''0.12.0''',
            '''torchaudio''': '''0.11.0''',
            '''python''': 3.9,
            '''cuda''': '''cu113''',
            '''install''': (
                '''python3 -m pip install --no-cache-dir -U torch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0'''
                ''' --extra-index-url https://download.pytorch.org/whl/cu113'''
            ),
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        '''1.10''': {
            '''torch''': '''1.10.2''',
            '''torchvision''': '''0.11.3''',
            '''torchaudio''': '''0.10.2''',
            '''python''': 3.9,
            '''cuda''': '''cu113''',
            '''install''': (
                '''python3 -m pip install --no-cache-dir -U torch==1.10.2 torchvision==0.11.3 torchaudio==0.10.2'''
                ''' --extra-index-url https://download.pytorch.org/whl/cu113'''
            ),
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        # torchaudio < 0.10 has no CUDA-enabled binary distributions
        '''1.9''': {
            '''torch''': '''1.9.1''',
            '''torchvision''': '''0.10.1''',
            '''torchaudio''': '''0.9.1''',
            '''python''': 3.9,
            '''cuda''': '''cu111''',
            '''install''': (
                '''python3 -m pip install --no-cache-dir -U torch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1'''
                ''' --extra-index-url https://download.pytorch.org/whl/cu111'''
            ),
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
    },
    '''tensorflow''': {
        '''2.11''': {
            '''tensorflow''': '''2.11.1''',
            '''install''': '''python3 -m pip install --no-cache-dir -U tensorflow==2.11.1''',
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        '''2.10''': {
            '''tensorflow''': '''2.10.1''',
            '''install''': '''python3 -m pip install --no-cache-dir -U tensorflow==2.10.1''',
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        '''2.9''': {
            '''tensorflow''': '''2.9.3''',
            '''install''': '''python3 -m pip install --no-cache-dir -U tensorflow==2.9.3''',
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        '''2.8''': {
            '''tensorflow''': '''2.8.2''',
            '''install''': '''python3 -m pip install --no-cache-dir -U tensorflow==2.8.2''',
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        '''2.7''': {
            '''tensorflow''': '''2.7.3''',
            '''install''': '''python3 -m pip install --no-cache-dir -U tensorflow==2.7.3''',
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        '''2.6''': {
            '''tensorflow''': '''2.6.5''',
            '''install''': '''python3 -m pip install --no-cache-dir -U tensorflow==2.6.5''',
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
        '''2.5''': {
            '''tensorflow''': '''2.5.3''',
            '''install''': '''python3 -m pip install --no-cache-dir -U tensorflow==2.5.3''',
            '''base_image''': '''nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04''',
        },
    },
}


if __name__ == "__main__":
				lowerCAmelCase__ :			Optional[int]         =argparse.ArgumentParser('Choose the framework and version to install')
				parser.add_argument(
				    '--framework', help='The framework to install. Should be `torch` or `tensorflow`', type=str, required=True
				)
				parser.add_argument('--version', help='The version of the framework to install.', type=str, required=True)
				lowerCAmelCase__ :			Any         =parser.parse_args()

				lowerCAmelCase__ :			Tuple         =past_versions_testing[args.framework][args.version]

				os.system(F"""echo \"export INSTALL_CMD=\'{info["install"]}\'\" >> ~/.profile""")
				print(F"""echo \"export INSTALL_CMD=\'{info["install"]}\'\" >> ~/.profile""")

				lowerCAmelCase__ :			str         =''''''
				if args.framework == "pytorch":
								lowerCAmelCase__ :			int         =info['''cuda''']
				os.system(F"""echo \"export CUDA=\'{cuda}\'\" >> ~/.profile""")
				print(F"""echo \"export CUDA=\'{cuda}\'\" >> ~/.profile""")









from __future__ import annotations

from collections import deque
from queue import Queue
from timeit import timeit

lowerCAmelCase__ :			str         ={
    'A': ['B', 'C'],
    'B': ['A', 'D', 'E'],
    'C': ['A', 'F'],
    'D': ['B'],
    'E': ['B', 'F'],
    'F': ['C', 'E'],
}






def 	a__ (							A__,			A__							):
			SCREAMING_SNAKE_CASE_		:    Optional[int] 		=				{start}
			SCREAMING_SNAKE_CASE_		:    int 		=				[start]
			SCREAMING_SNAKE_CASE_		:    Queue 		=				Queue()
			queue.put(A__							)
			while not queue.empty():
						SCREAMING_SNAKE_CASE_		:    Optional[Any] 		=				queue.get()
						for w in graph[v]:
									if w not in explored:
												explored.add(A__							)
												result.append(A__							)
												queue.put(A__							)
			return result






def 	a__ (							A__,			A__							):
			SCREAMING_SNAKE_CASE_		:    Any 		=				{start}
			SCREAMING_SNAKE_CASE_		:    str 		=				[start]
			SCREAMING_SNAKE_CASE_		:    List[Any] 		=				deque([start]							)
			while queue:
						SCREAMING_SNAKE_CASE_		:    Optional[Any] 		=				queue.popleft()
						for child in graph[v]:
									if child not in visited:
												visited.add(A__							)
												result.append(A__							)
												queue.append(A__							)
			return result






def 	a__ (							A__							):
			SCREAMING_SNAKE_CASE_		:    List[str] 		=				F'''from __main__ import G, {name}'''
			SCREAMING_SNAKE_CASE_		:    Tuple 		=				1_0_0_0_0
			SCREAMING_SNAKE_CASE_		:    Union[str, Any] 		=				timeit(F'''{name}(G, \'A\')''',			setup=A__,			number=A__							)
			print(F'''{name:<35} finished {number} runs in {res:.5f} seconds'''							)


if __name__ == "__main__":
				import doctest

				doctest.testmod()

				benchmark_function('breadth_first_search')
				benchmark_function('breadth_first_search_with_deque')
				# breadth_first_search                finished 10000 runs in 0.20999 seconds
				# breadth_first_search_with_deque     finished 10000 runs in 0.01421 seconds





