Utilizar GPU em vez de CPU nos treinamentos

[Mateus Scheper]

Olá!

Ao rodar meus treinos, vejo minha CPU indo para 100% e minha GPU em 0%, o que mostra não estar utilizando a GPU. Como alterar isso?
Ao rodar meus códigos, vejo essas mensagens abaixo, indicando erro ou algo faltando:

 

2023-02-25 17:05:44.249001: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library ‘cudart64_110.dll’; dlerror: cudart64_110.dll not found
2023-02-25 17:05:44.249235: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
2023-02-25 17:06:05.674979: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library ‘nvcuda.dll’; dlerror: nvcuda.dll not found
2023-02-25 17:06:05.675297: W tensorflow/stream_executor/cuda/cuda_driver.cc:263] failed call to cuInit: UNKNOWN ERROR (303)
2023-02-25 17:06:05.681264: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:169] retrieving CUDA diagnostic information for host: mateus-pc
2023-02-25 17:06:05.681596: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:176] hostname: mateus-pc
2023-02-25 17:06:05.682035: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.

 

Vendo no Google, o CUDA parece ser só para placas de vídeo NVIDIA, enquanto a minha é uma AMD RX 5700XT.
Fui ver também para instalar o tensorflow-gpu, mas na lista de instalação havia a mensagem “Removido. Instale o tensorflow em vez disso”, indicando que o pacote não é mais utilizado.

[Denny Ceccon]

Olá Mateus,

O Tensorflow agora identifica automaticamente a GPU e usa quando ela está disponível, mas você precisa instalar drivers específicos para que este reconhecimento seja feito. Cada placa tem um processo diferente, se você quiser usar sua GPU o melhor caminho é procurar ajuda no Google mesmo.

[Autor]

Posts