• DaniAndTheWeb's avatar
    Automatic torch install for amd on linux · a407c9f0
    DaniAndTheWeb authored
    This commit allows the launch script to automatically download rocm's torch version for AMD GPUs using an external GPU detection script. It also prints the operative system and GPU in use.
    a407c9f0
launch.py 12.6 KB