簡體   English   中英

使用Cuda 10安裝pip tensorflow-gpu后出現錯誤

[英]Error after installing pip tensorflow-gpu with cuda 10

如果tensorflow-gpu在cpu上運行錯誤代碼,我只想像conda版本那樣使用tensorflow的pip版本。

在我導入tensorflow時為ubuntu 18.0.4安裝cuda 10和cudnn后,它給我以下錯誤。

PS:我清除了與cuda和Nvidia有關的所有內容,然后安裝了新的Cuda 10。

Python 3.6.7 | Anaconda自定義(64位)| (默認值,2018年10月23日,19:16:44)Linux上的[GCC 7.3.0]鍵入“幫助”,“版權”,“信用”或“許可證”以獲取更多信息。

>>>導入張量流

追溯(最近一次通話):文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py”,行58,來自tensorflow.python.pywrap_tensorflow_internal導入*文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py”,第28行,在_pywrap_tensorflow_internal = swig_import_helper()文件中,“ / home / purav / anaconda3 / envs / purav / lib / python3.6 / site-packages / tensorflow / python / pywrap_tensorflow_internal.py“,第24行,在swig_import_helper _mod = imp.load_module('_ pywrap_tensorflow_internal',fp,路徑名,描述)文件” / home /purav/anaconda3/envs/purav/lib/python3.6/imp.py“,第243行,在load_module中返回load_dynamic(名稱,文件名,文件)文件“ / home / purav / anaconda3 / envs / purav / lib / python3 .6 / imp.py“,行343,在load_dynamic中返回_load(spec)ImportError:libcublas.so.9.0:無法打開共享庫文件:沒有這樣的文件或目錄

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

追溯(最近一次通話最后一次):文件“ /”,第1行,文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/ init .py”,第24行,從tensorflow.python import pywrap_tensorflow#pylint:disable = unused-import File“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/ init .py”,第49行,在從tensorflow.python導入pywrap_tensorflow文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py”,行74,在抬高ImportError(msg)ImportError:Traceback (最近一次通話):文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py”,行58,來自tensorflow.python.pywrap_tensorflow_internal import *文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py”,第28行,在_pywrap_tensorflow_internal = swig_import_helper()中,文件“ / home / purav / anaconda3 /envs/purav/lib/python3.6/sit e-packages / tensorflow / python / pywrap_tensorflow_internal.py“,第24行,在swig_import_helper _mod = imp.load_module('_ pywrap_tensorflow_internal',fp,路徑名,描述)文件” / home / purav / anaconda3 / envs / purav / lib / python3 .6 / imp.py“,第243行,在load_module中返回load_dynamic(名稱,文件名,文件),文件“ /home/purav/anaconda3/envs/purav/lib/python3.6/imp.py”,第343行,在load_dynamic返回_load(spec)ImportError:libcublas.so.9.0:無法打開共享庫文件:沒有這樣的文件或目錄
ImportError:libcublas.so.9.0:無法打開共享庫文件:沒有這樣的文件或目錄

我不知道為什么會出現此錯誤,我沒有cuda verison 9

可通過pip獲得的tensorflow-gpu的當前穩定版本是tensorflow-gpu ,它是針對CUDA 9.0構建的。 如果您查看tensorflow github存儲庫的發行號22706 ,則即將基於CUDA 10構建即將發布的tensorflow-gpu v1.13的官方二進制文件。

自2019年1月23日起,已針對CUDA 10構建了tensorflow-gpu v1.13rc0版本 。可以通過指定以下版本使用pip進行安裝:

點安裝tensorflow-gpu == 1.13.0rc0

由於系統中的CUDA 9已被CUDA 10取代,因此默認的tensorflow-gpu代碼失敗,因為它的版本為1.12,並期望系統中存在CUDA 9二進制文件。 因此,當前,如果您想在CUDA 10中使用tensorflow,那么最快的解決方案就是按照上面指定的方式升級tensorflow版本。

問題是您尚未完全滿足TensorFlow GPU要求。
完整名單是這里的TF網站上。

錯誤的主要原因似乎是您沒有使用cuDNN SDK正確配置CUDA工具包9.0 ,請注意您需要的版本大於或等於7.2,並將其添加到系統路徑中。

我正在運行python 3.7並能夠通過運行cuda 10安裝tensorflow:

 pip install --no-cache-dir https://github.com/evdcush/TensorFlow-wheels/releases/download/tf-1.12.0-py37-gpu-10.0/tensorflow-1.12.0-cp37-cp37m-linux_x86_64.whl

暫無
暫無

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

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