我正在尝试在离线环境(windows 64)中使用tensorflow-gpu库。我的python版本为3.6,为64位,我使用的是Anaconda3解释器(5.2.0版本)。库达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安装安装了这些文件。我看到库出现在“C:\ProgramData\Anaconda3 3\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;我还试图安装tensorflow-gpu-1.10与CUDA 9.2和CuDNN 7.1.4,我收到了同样的错误。
发布于 2018-09-12 11:34:51
尝试使用CUDA版本9.0代替,它是明确声明在Tensorflow安装指南中,9.0是必需的。降级到早期版本为我解决了这个问题。
发布于 2018-09-12 13:22:28
您安装了tensorflow 1.1,这需要CUDA8.0和CuDNN 5.1才能在GPU上运行。
https://stackoverflow.com/questions/52293578
复制相似问题