Hi
I'm having difficulty setting this environment variable (embarrassingly).
So, I have the same problem, probably because I'm using an RTX2060. After starting training:
704953: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
The solution appears to be to set the environment variable to allow growth.
I've spent a long time looking at the conda documentation, but this command (from docs.conda.io)
conda env config vars set my_var=value is failing to work
So I activated the faceswap environment, then entered in terminal:
conda env config vars set TF_FORCE_GPU_ALLOW_GROWTH=true
but this error is returned:
usage: conda-env [-h] {create,export,list,remove,update} ...
conda-env: error: invalid choice: 'config' (choose from 'create', 'export', 'list', 'remove', 'update')
but its the command on the conda website!!!!
I've also tried setting the variable in the _config.yml file, but still get the error when starting training
Can anyone help?