Tensorflow 2.4.1 - 无法调用 ptxas.exe

Question

我尝试在 GPU 支持（GTX 1660 SUPER）下运行 Tensorflow。

我使用 anaconda 创建了一个环境，然后安装了cudatoolkit（版本11.0.221）和tensorflow-gpu（版本2.4.1）。之后，我下载了cuDNN（版本8.0.4），并将所有文件从 cuDNN 的bin文件夹复制到我的环境的 bin 文件夹（位于 ）anaconda3\envs\<env name>\Library\bin。在我的脚本中，我使用 来设置 GPU 内存的内存限制tf.config.experimental.set_memory_growth。

当我运行脚本（使用卷积算法）时，我收到一条警告，指出Couldn't invoke ptxas.exe --version出现错误后出现的情况Call to CreateProcess failed. Error code: 2。启动失败后，我得到：Relying on driver to perform ptx compilation. Modify $PATH to customize ptxas location.

我已经尝试切换到 cuDNN 版本8.1.1。

我该如何解决这个问题？

Answer 1

我对此有了新的修复。

首先我尝试使用tensorflow=2.3, cudnn=7.6.5 and cudatoolkit=10.1之前答案中提到的方法。然而，每次我训练模型时，过程都会变得陈旧，并且训练似乎停留在第 1 轮。

然后，我通过运行设法将 ptxas 包含在我的 conda 环境中conda install -c nvidia cuda-nvcc我正在使用的软件包是：

tensorflow=2.9, cudnn=8.1.0, cudatoolkit=11.2.2, cuda-nvcc=11.7.99 and python=3.9

我现在在 Windows 10 上运行一切都完美。

Answer 2

为了社区的利益添加@Zuk Levinson 评论

通过使用解决了这个问题

tensorflow=2.3, cudnn=7.6.5 and cudatoolkit=10.1