【发布时间】:2019-02-17 00:00:56
【问题描述】:
我正在尝试在离线环境(Windows 64)中使用 tensorflow-gpu 库。 我的 python 版本是 3.6 的 64 位,我使用的是 Anaconda3 解释器(5.2.0 版本)。 Cuda 9.2 和 cuddn 7.1.4(从https://developer.nvidia.com/rdp/cudnn-archive 下载)
我下载了 tensorflow-1.2.1-py36_0.tar.bz2 和 tensorflow-gpu-1.1.0-np112py36_0.tar.bz2 并使用 Conda install 安装了文件。 我看到库出现在“C:\ProgramData\Anaconda3\Lib\site-packages”下,但是虽然 tensorflow(在 cpu 下)运行良好,但在尝试运行 tensorflow-gpu 时出现错误 "_pywrap_tensorflow_internal" 。
环境变量:
CUDA_HOME: c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2
CUDA_PATH: c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2
CUDA_PATH_V9_2: c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2
路径:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\CUPTI\libx64;c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\lib\x64;c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin;c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\libnvvp;
我也尝试使用 CUDA 9.2 和 CuDNN 7.1.4 安装 tensorflow-gpu-1.10,我收到了同样的错误。
【问题讨论】:
标签: python tensorflow deep-learning gpu