[Python] How to Select/Isolate GPUs when Using ROCm in Python

Discussão em 'Python' iniciado por Stack, Fevereiro 9, 2025.

  1. Stack

    Stack Membro Participativo

    Is their an equivalent for os.environ["CUDA_VISIBLE_DEVICES"]="0" in the os module for isolating GPUs when using AMD ROCm?

    Continue reading...
    Stack, Fevereiro 9, 2025
    #1

Compartilhe esta Página