问题

无法调用 ptxas.exe


我尝试使用 GPU 支持运行 Tensorflow(GTX 1660 SUPER)。

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

当我运行脚本(使用卷积算法)时,我收到一条警告说 Couldn't invoke ptxas.exe --version 这是在 Call to CreateProcess 失败之后出现的。错误代码:2错误。启动失败后,我得到:依赖驱动执行 ptx 编译。修改 $PATH 以自定义 ptxas 位置。

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

我该如何解决这个问题?

推荐答案

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

通过使用

tensorflow=2.3, cudnn=7.6.5 and cudatoolkit=10.1