[Docker] Use Torch GPU on gpu device#17676

yongwww merged 1 commit intoapache:mainapache/tvm:mainfrom

yongwww:use_torch_gpuyongwww/tvm:use_torch_gpuCopy head branch name to clipboard