es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Tag: GPYTORCH

Actualice CUDNN a 8.2 en google colab.

Quiero actualizar la versión de CUDNN de 8.0 a 8.1 y la versión de CUDA a 11.2, pero no estoy seguro de cómo hacerlo en Colab. A continuación, se muestra el script que escribí para ejecutar en Google Colab, pero no cambia nada. ¿Qué cambios debo hacer aquí? ¿O hay . . . Read more

¿Cómo usar los núcleos de tensor en pytorch y tensorflow?

Estoy utilizando una GPU Nvidia RTX con núcleos tensoriales, quiero asegurarme de que pytorch/tensorflow está utilizando sus núcleos tensoriales. He notado en algunos artículos que los núcleos tensoriales se utilizan para procesar flotante16 y por defecto pytorch/tensorflow utiliza flotante32. Han introducido una librería que hace “entrenamiento de precisión mixta y . . . Read more

¿Cómo cambio mi propio conjunto de datos al formato de conjunto de datos MNIST?

data_transform_test = transforms.Compose([transforms.ToTensor(), transforms.Grayscale(num_output_channels=1), transforms.Resize((28, 28)), transforms.Normalize((0.5,), (0.5,)) ]) Esto es lo que intenté para mi implementación de un modelo basado en el conjunto de datos MNIST. No puedo cambiar la arquitectura completa del modelo porque eso es demasiado trabajo para mí. Pero mi código me está dando este error. . . . Read more