[tune] Fix tutorial training on GPU (#12914)
[tune] Fix tutorial training on GPU (#12914)


pcmoritz/ray-1,pcmoritz/ray-1,ray-project/ray,ray-project/ray,pcmoritz/ray-1,pcmoritz/ray-1,pcmoritz/ray-1,pcmoritz/ray-1,ray-project/ray,ray-project/ray,ray-project/ray,pcmoritz/ray-1,ray-project/ray,pcmoritz/ray-1,ray-project/ray,ray-project/ray