Tensorflow 2.4.1 -无法调用ptxas.exe

Question

我尝试运行Tensorflow与GPU支持(GTX 1660超级)。

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

当我运行脚本(使用卷积算法)时，我会得到一个警告：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，正如前面的答案中提到的那样。然而，每次我提出一个模型来训练，这个过程已经过时了，训练似乎陷入了一个时代。

然后，通过运行conda install -c nvidia cuda-nvcc，我成功地在conda环境中包含了ptxas，我使用的包如下：

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