Anaconda 显示此错误，无法正确训练模型

Question

2021-02-03 19:46:53.571084: W tensorflow/stream_executor/gpu/redzone_allocator.cc:314] Internal: Invoking GPU asm compilation is supported on Cuda non-Windows platforms only
Relying on driver to perform ptx compilation.
Modify $PATH to customize ptxas location.
This message will be only logged once.

当我尝试训练深度神经网络时，出现上述错误。它不会停止训练，但第一次迭代无法正常工作。见下文：

Epoch 1/10
   1/1875 [..............................] - ETA: 0s - loss: 2.3001 - accuracy: 0.1250
WARNING:tensorflow:Callbacks method `on_train_batch_end` is slow compared to the batch time (batch time: 0.0000s vs `on_train_batch_end` time: 0.0116s). Check your callbacks.

帮我修复这个错误。

我使用的是tensorflow版本2.3.0，cuda工具包版本10.1.243，cudnn版本7.6.5，我的电脑有NVDIA GEFORCE RTX 2070，采用Super Max-Q设计。

有些解决方案说降级tensorflow版本，但对于某些项目我需要2.3.0，所以请任何人解决这个错误。

Answer 1

我遇到了同样的问题，我的TensorFlow版本是2.4.1。使用 Python 3.9 和 cuda 10.1。幸运的是，我通过安装以下软件包找到了解决方案，

conda install -c conda-forge cudatoolkit-dev

原始链接在这里。