74 votos

¿Cómo comprobar si pytorch está usando la GPU?

Me gustaría saber si está usando mi GPU. Es posible detectar con si hay alguna actividad de la GPU durante el proceso, pero quiero algo escrito en un `` script.

¿Hay alguna manera de hacerlo?

118voto

vinzee Puntos 2714

Esto va a funcionar :

Esto me dice que la GPU está siendo utilizada por .

10voto

TimeSeam Puntos 361

En el sitio de la oficina y la página de inicio get, compruebe la GPU para PyTorch como se muestra a continuación:

Referencia: PyTorch Obtener inicio

Iteramos.com

Iteramos es una comunidad de desarrolladores que busca expandir el conocimiento de la programación mas allá del inglés.
Tenemos una gran cantidad de contenido, y también puedes hacer tus propias preguntas o resolver las de los demás.

Powered by:

X