gpt4 book ai didi

python - 我没有 Nvidia GPU,但想在 CPU 上运行 Tensorflow 模型。为什么它一直要求一些 CUDA DLL?

转载 作者:行者123 更新时间:2023-12-04 15:42:43 26 4
gpt4 key购买 nike

我关注了these instructions

具体来说,我想运行从 Github 下载的 Tensorflow 模型。我的电脑上只有一个 Intel GPU,所以我想在我的 CPU 上执行 Tensorflow 模型。如所述here on GitHub ,应该可以通过将 use-gpu 参数设置为 false 来实现。所以我运行这个命令:

python test_model.py model=iphone_orig dped_dir=dped/ test_subset=full iteration=all resolution=orig use_gpu=false

但是,我收到以下错误,最后两行表明 tensorflow 尝试在 GPU 上运行,这是控制台窗口:

C:\Users\username\Downloads\DPED-master\DPED-master>python test_model.py model=iphone_orig dped_dir=dped/ test_subset=full iteration=all resolution=orig use_gpu=false
Traceback (most recent call last):
File "C:\Users\username\Downloads\WPy64-3720\python-3.7.2.amd64\lib\site-packages\tensorflow\python\platform\self_check.py", line 62, in preload_check
ctypes.WinDLL(build_info.nvcuda_dll_name)
File "C:\Users\username\Downloads\WPy64-3720\python-3.7.2.amd64\lib\ctypes\__init__.py", line 356, in __init__
self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] Das angegebene Modul wurde nicht gefunden

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "test_model.py", line 5, in <module>
import tensorflow as tf
File "C:\Users\username\Downloads\WPy64-3720\python-3.7.2.amd64\lib\site-packages\tensorflow\__init__.py", line 28, in <module>
from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import
File "C:\Users\username\Downloads\WPy64-3720\python-3.7.2.amd64\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
from tensorflow.python import pywrap_tensorflow
File "C:\Users\username\Downloads\WPy64-3720\python-3.7.2.amd64\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>
self_check.preload_check()
File "C:\Users\username\Downloads\WPy64-3720\python-3.7.2.amd64\lib\site-packages\tensorflow\python\platform\self_check.py", line 70, in preload_check
% build_info.nvcuda_dll_name)
ImportError: Could not find 'nvcuda.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Typically it is installed in 'C:\Windows\System32'. If it is not present, ensure that you have a CUDA-capable GPU with the correct driver installed.

你找到相关的test_model.py文件here

我尝试了几次执行,有和没有 GPU 设置。我该怎么做才能修复它?

最佳答案

tensorlfow有两个模块:'tensorflow','tensorflow-gpu'在 cpu 上你需要使用 pip install tensorflow 安装 tensorlfow 或者在 conda 上 conda install tensorflow

编辑第二个问题:

如果将 TensorFlow 操作放在 GPU 上,则执行引擎必须具有该操作的 GPU 实现,称为内核。
如果内核不存在,则放置会导致运行时错误。此外,如果请求的 GPU 设备不存在,则会引发运行时错误。
如果请求 GPU 设备导致错误,最好的处理方式是允许将操作放在 CPU 上。

一个答案是删除所有 GPU 配置,第二个是软放置,如果找不到 GPU,如上文所述使用 config.allow_soft_placement = True

关于python - 我没有 Nvidia GPU,但想在 CPU 上运行 Tensorflow 模型。为什么它一直要求一些 CUDA DLL?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57207936/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com