在Window-7 x64下的离线环境中运行Tensorflow-GPU的问题

时间:2018-09-12 11:00:55

标签: python tensorflow deep-learning gpu

我正在尝试在离线环境(Windows 64)中使用tensorflow-gpu库。 我的python版本是3.6 for 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,但收到了相同的错误。

2 个答案:

答案 0 :(得分:0)

请尝试改用CUDA版本9.0,Tensorflow installation guide中明确指出需要9.0。降级到较早的版本对我来说解决了这个问题。

答案 1 :(得分:0)

您安装了tensorflow 1.1,它需要CUDA 8.0和CuDNN 5.1才能在GPU上运行。