报错:tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.
中看到了两种解决办法,但是由于在 TensorFlow 2 中,tf.GPUOptions 和 tf.ConfigProto 已经被弃用,chatGPT推荐使用 tf.config.experimental 模块中的 API 进行 GPU 配置。ChatGPT一直和我说是CUDA和cuDNN的版本问题,但是上网搜索了TensorFlow与之对应的版本,发现没有问题。
原创
2023-04-11 11:23:02 ·
670 阅读 ·
1 评论