Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for gpu_platform_count (0.11 sec)

  1. configure.py

      # ROCm / CUDA are mutually exclusive.
      # At most 1 GPU platform can be configured.
      gpu_platform_count = 0
      if environ_cp.get('TF_NEED_ROCM') == '1':
        gpu_platform_count += 1
      if environ_cp.get('TF_NEED_CUDA') == '1':
        gpu_platform_count += 1
      if gpu_platform_count >= 2:
        raise UserInputError('CUDA / ROCm are mututally exclusive. '
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Oct 02 22:16:02 UTC 2024
    - 48.2K bytes
    - Viewed (0)
Back to top