簡體   English   中英

安裝了CUDA 9.2時無法導入Tensorflow 1.10

[英]Unable to import tensorflow 1.10 with CUDA 9.2 installed

我想使用tensorflow GPU版本。 我已經通過pip install命令安裝了tensorflow,它安裝了最新版本1.10。 觀看視頻后,我安裝了CUDA工具包,最新版本為9.2,而cudnn版本為7.2.1,適用於CUDA 9.2。 導入時出現錯誤

追溯(最近一次通話):文件“ C:\\ Users \\ Lenovo \\ AppData \\ Local \\ Programs \\ Python \\ Python36 \\ lib \\ site-packages \\ tensorflow \\ python \\ platform \\ self_check.py”,行75,在preload_check ctypes中.WinDLL(build_info.cudart_dll_name)文件``C:\\ Users \\ Lenovo \\ AppData \\ Local \\ Programs \\ Python \\ Python36 \\ lib \\ ctypes__init __。py'', init self._handle = _dlopen(self._name,mode) OSError:[WinError 126]找不到指定的模塊

在處理上述異常期間,發生了另一個異常:

追溯(最近一次調用):導入張量流中的文件“”,行1,文件C:\\ Users \\ Lenovo \\ AppData \\ Local \\ Programs \\ Python \\ Python36 \\ lib \\ site-packages \\ tensorflow__init __。py“,行22 ,從tensorflow.python import pywrap_tensorflow#pylint:disable =未使用的導入文件“ C:\\ Users \\ Lenovo \\ AppData \\ Local \\ Programs \\ Python \\ Python36 \\ lib \\ site-packages \\ tensorflow \\ python__init __。py”,第49行,從tensorflow.python中導入pywrap_tensorflow文件“ C:\\ Users \\ Lenovo \\ AppData \\ Local \\ Programs \\ Python \\ Python36 \\ lib \\ site-packages \\ tensorflow \\ python \\ pywrap_tensorflow.py”,行30,位於self_check.preload_check( )文件“ C:\\ Users \\ Lenovo \\ AppData \\ Local \\ Programs \\ Python \\ Python36 \\ lib \\ site-packages \\ tensorflow \\ python \\ platform \\ self_check.py”,第82行,位於preload_check%(build_info.cudart_dll_name,build_info。 cuda_version_number))ImportError:找不到'cudart64_90.dll'。 TensorFlow要求將此DLL安裝在%PATH%環境變量中命名的目錄中。 從以下URL下載並安裝CUDA 9.0: https : //developer.nvidia.com/cuda-90-download-archive

無法理解為什么要使用較舊的版本cudart64_90。 我必須安裝CUDA 9.0還是可以在現有安裝中進行一些更改。

CUDA 9.2還不支持TF 1.10預制安裝程序。 您可以卸載使用預安裝的二進制文件安裝的TF 1.10,然后為Cuda 9.2重建TF。

我只是按照下面的鏈接從Cuda 9.2的源代碼構建TF:

https://www.python36.com/how-to-install-tensorflow-gpu-with-cuda-9-2-for-python-on-ubuntu/2/

上面的說明很棒並且有效。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM