Meu ambiente é: Win10 cuda11.0 cudnn8.0 Python3.7 tensorflow2.4.0
A execução do código relata um erro: Falha na chamada para CreateProcess. Código do erro: 2
Encontrei uma solução em um site em inglês (usei o método um para resolver o problema)
método um:
Você pode garantir que ptxas esteja em seu ambiente conda executando conda install -c nvidia cuda-nvcc , que funcionou para mim.
conda install -c nvidia cuda-nvcc
Executei o comando acima na linha de comando do ambiente virtual usado para executar o código e resolvi o problema com sucesso.
Método Dois
Tive um problema semelhante depois de instalar apenas a parte Runtime do Cudatoolkit (usando o instalador da GUI no Windows).
Certifique-se de instalar todo o pacote CUDA (incluindo tempo de execução e desenvolvimento). Depois disso, o ptxas.exe deve aparecer na pasta bin do CUDA e o Tensorflow pode encontrá-lo.
Além disso, se você encontrar o seguinte erro
Não está criando dispositivos XLA, tf_xla_enable_xla_devices não definido