[PyTorch] How to check which GPU device our data used
Last Updated on 2021-07-05 by Clay
When I using PyTorch to train a model, I often use GPU_A to train the model, save model. But if I load the model I saved to test some new data, I always put the new data in a different GPU, we called it GPU_B.
Read More »[PyTorch] How to check which GPU device our data used