我尝试使用 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