I tried to run example/cats_and_dogs.inpynb on my V100 GPU
Data and models are in the GPU memorys
But the usage of GPU is always 0%, the other way, CPU's usage are in 900%~1000%
I am not sure what device am i training on
p.s. I tried to print the ViT model device ViT.device, it tells me device is not define
I tried to run example/cats_and_dogs.inpynb on my V100 GPU Data and models are in the GPU memorys But the usage of GPU is always 0%, the other way, CPU's usage are in 900%~1000% I am not sure what device am i training on
p.s. I tried to print the ViT model device
ViT.device
, it tells me device is not define