


'''simple docstring'''

import argparse
import os


lowerCamelCase							:       Tuple  						=     {
    '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							:       List[str]  						=     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							:       Optional[int]  						=     parser.parse_args()

 lowerCamelCase							:       Any  						=     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							:       List[Any]  						=     ''
 if args.framework == "pytorch":
  lowerCamelCase							:       Union[str, Any]  						=     info['cuda']
 os.system(f"""echo \"export CUDA='{cuda}'\" >> ~/.profile""")
 print(f"""echo \"export CUDA='{cuda}'\" >> ~/.profile""")








'''simple docstring'''

import math
import os

import cva
import numpy as np

lowerCamelCase							:       Optional[Any]  						=     2_5_5.0

def _SCREAMING_SNAKE_CASE				(A  ,			A	)	->	float:






       """simple docstring"""







       lowercase__				      =			np.mean((original - contrast) ** 2	)
       if mse == 0:
              return 100

       return 20 * math.logaa(PIXEL_MAX / math.sqrt(A	)	)

def _SCREAMING_SNAKE_CASE				()	->	None:






       """simple docstring"""







       lowercase__				      =			os.path.dirname(os.path.realpath(A	)	)
       # Loading images (original image and compressed image)
       lowercase__				      =			cva.imread(os.path.join(A  ,			'''image_data/original_image.png'''	)	)
       lowercase__				      =			cva.imread(os.path.join(A  ,			'''image_data/compressed_image.png'''	)  ,			1	)

       lowercase__				      =			cva.imread(os.path.join(A  ,			'''image_data/PSNR-example-base.png'''	)	)
       lowercase__				      =			cva.imread(
           os.path.join(A  ,			'''image_data/PSNR-example-comp-10.jpg'''	)  ,			1	)

       # Value expected: 29.73dB
       print('''-- First Test --'''	)
       print(f"PSNR value is {peak_signal_to_noise_ratio(A  ,			A	)} dB"	)

       # # Value expected: 31.53dB (Wikipedia Example)
       print('''\n-- Second Test --'''	)
       print(f"PSNR value is {peak_signal_to_noise_ratio(A  ,			A	)} dB"	)


if __name__ == "__main__":
 main()




