小编Zuk*_*son的帖子

Tensorflow 2.4.1 - 无法调用 ptxas.exe

我尝试在 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

我该如何解决这个问题?

python anaconda tensorflow

8
推荐指数
2
解决办法
1万
查看次数

标签 统计

anaconda ×1

python ×1

tensorflow ×1