为使用faceswap玩一下换脸,在win10+rtx2070平台上搭建TensorFlow-gpu 1.13.1环境。安装完毕后import tensorflow没有问题,但在运行时总是报Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED错误,上网多方查证后认为cuda、cudnn、TensorFlow的版本搭配没有问题,但就是反复重装环境也不行。。
最后,注销一下用户,再登陆进去,就!好!了! 所以遇到同样问题时你可以尝试注销或重启下电脑,网上有个别人也提到了这个解决方法。
软硬件详细版本如下:
gpu_cuda: 10.0
gpu_cudnn: 7.4.1
gpu_devices: GPU_0: GeForce RTX 2070, GPU_1: GeForce RTX 2070
gpu_driver: 417.35
os_platform: Windows-10-10.0.17763-SP0
py_version: 3.6.2