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.

Python3 eliminación de advertencia de antorcha.

Tengo un script de python que usa torch y easyocr.
Cuando lo ejecuto, obtengo en la terminal mi salida pero también veo dos advertencias:

CUDA no está disponible, se usa la CPU. Nota: Este módulo es mucho más rápido con una GPU.
[W NNPACK.cpp:79] No se pudo inicializar NNPACK! Razón: Hardware no soportado.

¿Hay alguna forma de agregar algo en mi código de python para suprimir esas dos advertencias?
Gracias.

Intenté lo siguiente pero no funcionó:

#!/usr/bin/env python3

– coding: utf-8 –

import cv2
import numpy as np
import easyocr
import sys
import requests
import os

os.environ[“CUDA_VISIBLE_DEVICES”]=””

Gracias.

Tags:  ,

Answer

  1. Avatar for davy.ai

    Es posible suprimir las advertencias configurando la variable de entorno CUDA_VISIBLE_DEVICES con una cadena vacía. Esto se puede hacer en su código de Python como se ha hecho anteriormente.

    os.environ["CUDA_VISIBLE_DEVICES"]=""
    

Comments are closed.