【发布时间】:2020-07-26 06:11:57
【问题描述】:
我一直在尝试让 CUDA 与 TensorFlow 一起工作一段时间,因为我一直在构建的神经网络现在需要数小时才能在我的 CPU 上进行训练,如果能获得如此大的速度提升,那就太好了.但是,每当我尝试将它与 TensorFlow 一起使用时(它可以与 PyTorch 一起使用,但我想学习多个 API),它会告诉我运行 CUDA 所需的 .dll 文件之一不存在,而实际上它确实存在。
我已经从 dll-files.com 下载并用其他版本替换了该 .dll。我已经尝试卸载并重新安装 TensorFlow、CUDA 和 cuDNN。我尝试了不同版本的 CUDA,但这只会导致找不到所有 .dll 文件(是的,我确实更改了 CUDA_PATH 值)。我尝试在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0 和 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin 之间切换路径,看看是否有任何改变。
如果有人能提供帮助,将不胜感激。
【问题讨论】:
标签: python-3.x dll tensorflow2.0