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