TensorFlow 2.10 com GPU, Python 3.10, SciKit-Learn, Spyder e mais alguns pacotes


Apenas atualizando a “receita” do artigo TensorFlow 2.10 no Windows com suporte a GPU via conda+pip + TensorFlow-Addons e Spyder. O guia do TensorFlow coloca o Python 3.9, mas não vejo razão para não colocar o Python 3.10, visto que ele é suportado pelo TensorFlow 2.10.

Além disso, acrescentei o pacote scikit-learn-intelex para acelerar o scikit-learn e também o cudatoolkit-dev, para evitar este erro com o ptxas:

2023-01-13 01:28:00.497968: I tensorflow/stream_executor/cuda/cuda_dnn.cc:384] Loaded cuDNN version 8201
2023-01-13 01:28:01.056956: E tensorflow/core/platform/windows/subprocess.cc:287] Call to CreateProcess failed. Error code: 2, command: '"ptxas.exe" "--version"'
2023-01-13 01:28:01.057635: E tensorflow/core/platform/windows/subprocess.cc:287] Call to CreateProcess failed. Error code: 2, command: '"ptxas.exe" "--version"'
2023-01-13 01:28:01.057654: W tensorflow/stream_executor/gpu/asm_compiler.cc:80] Couldn't get ptxas version string: INTERNAL: Couldn't invoke ptxas.exe --version

Agora a receita ficou assim:

conda create --name tf python=3.10
conda activate tf
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
pip install "tensorflow<2.11"
pip install tensorflow-addons[tensorflow]
conda install pandas scikit-learn pillow spyder scikit-learn-intelex
conda install -c conda-forge cudatoolkit-dev

 

5 1 voto
Article Rating
(Visitado 49 vezes, 1 visitas hoje)

Link permanente para este artigo: https://www.skooterblog.com/2023/01/16/tensorflow-2-10-com-gpu-python-3-10-scikit-learn-spyder-e-mais-alguns-pacotes/

Inscrever
Notificar sobre
guest

0 Comentários
Inline Feedbacks
Ver todos os comentários
0
Gostaríamos de saber o que você pensa, deixe seu comentáriox