Automatic launch argument for AMD GPUs
This commit adds a few lines to detect if the system has an AMD gpu and adds an environment variable needed for torch to recognize the gpu.master
parent
82725f0ac4
commit
cbf4b3472b
Loading…
Reference in New Issue