Press "Enter" to skip to content

Posts published in 2月 2021

AutoTVM 利用多显卡加速 Tuning

junyi 0

场景 双 NVIDIA 1080Ti, Linux 服务器 需求 AutoTVM 的 Tuning 可以让两张 CUDA 显卡同时运行模型 (AMD的没试过) 过程 当前场景为tuner和runner都跑在同一台机器上的情况 如果需要一台跑tuner,一台跑runner,请配置好 tracker 和 server 的 IP tracker 和 server 都可以放在有显卡的机器里 首先查看一下 GPU 的 id,你可以通过这条命令来查看: nvidia-smi -L 比如我这里输出的是 0 和…