python:无法导入tensorflow-gpu

时间:2018-01-02 12:10:50

标签: python tensorflow tensorflow-gpu

我在我的机器上使用Anaconda3成功创建了我的tensorflow环境,并在此link上介绍了步骤。但是,当我试图做的时候:

>>> import tensorflow as tf

我收到以下错误消息:OSError和ImportError。

Traceback (most recent call last):
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow\lib\site-
packages\tensorflow\python\platform\self_check.py", line 75, in preload_check
ctypes.WinDLL(build_info.cudart_dll_name)
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow\lib\ctypes\__init__.py", line 351, in __init__
self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] This specified module could not be found

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
from tensorflow.python import *
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
from tensorflow.python import pywrap_tensorflow
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>
self_check.preload_check()
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow\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_80.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 8.0 from this URL: https://developer.nvidia.com/cuda-toolkit

这可能与我的cpu有关吗?我使用Ryzen 1700。 + GPU:GTX 1050 Ti - 略逊于1080 ti ......但我听说GPU仍然比CPU更好。

1 个答案:

答案 0 :(得分:3)

您需要先安装CUDA库,我建议您按照以下链接进行操作:

https://gist.github.com/mjdietzx/0ff77af5ae60622ce6ed8c4d9b419f45

更新: 对于Windows操作系统,请遵循以下方法:

http://blog.nitishmutha.com/tensorflow/2017/01/22/TensorFlow-with-gpu-for-windows.html