在tensorflow环境搭建中不需要安装cuda吗

Question

在tensorflow环境搭建中不需要安装cuda吗

我运行tensorflow的检测时，则出现如下错误提示
D:\Envs\nlp\python.exe D:/python/chabot/data_utils.py
Traceback (most recent call last):
File “D:\Envs\nlp\lib\site-packages\tensorflow\python\platform\self_check.py”, line 75, in preload_check
ctypes.WinDLL(build_info.cudart_dll_name)
File “D:\Envs\nlp\lib\ctypes_init_.py”, line 348, in init
self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] 找不到指定的模块。

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File “D:/python/chabot/data_utils.py”, line 1, in
import tensorflow as tf
File “D:\Envs\nlp\lib\site-packages\tensorflow_init_.py”, line 24, in
from tensorflow.python import *
File “D:\Envs\nlp\lib\site-packages\tensorflow\python_init_.py”, line 49, in
from tensorflow.python import pywrap_tensorflow
File “D:\Envs\nlp\lib\site-packages\tensorflow\python\pywrap_tensorflow.py”, line 30, in
self_check.preload_check()
File “D:\Envs\nlp\lib\site-packages\tensorflow\python\platform\self_check.py”, line 82, in preload_check
% (build_info.cudart_dll_name, build_info.cuda_version_number))
ImportError: Could not find ‘cudart64_90.dll’. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 9.0 from this URL: https://developer.nvidia.com/cuda-toolkit

Process finished with exit code 1

Answer 1

Mr_Ricky 2018-10-16 15:50:14

首先你要确定你的电脑是否支持GPU，是否有GPU的显卡，如果有的话才能装cuda和cudnn，你的错误就是cuda和cudnn的版本有问题。需要找对应的版本。