簡體   English   中英

在ubuntu18.04中導入錯誤tensorflow-gpu

[英]Import error tensorflow-gpu in ubuntu18.04

導入張量流時出現一些錯誤。

我的電腦規格:

作業系統:ubuntu 18.04

英偉達RTX 2080 Ti * 2

Nvidia驅動程序415

CUDA:10.0

cuDNN:7.3.0張量流:1.11.0

import tensorflow

錯誤:

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

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

追溯(最近一次調用):文件“”,行1,位於文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/ init .py”,行22,來自tensorflow.python import pywrap_tensorflow #pylint:disable = unused-import文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/python/ init .py”,行49,來自tensorflow.python import pywrap_tensorflow文件“ / usr / local” /lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py“,行74,在抬高ImportError(msg)ImportError:追溯(最近一次調用):文件” / usr / local / lib / python3。 6 / dist-packages / tensorflow / python / pywrap_tensorflow.py“,第58行,來自tensorflow.python.pywrap_tensorflow_internal import *文件” /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal。 py”,_ pywrap_tensorflow_internal = swig_import_helper()中的第28行,文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py”,第24行,位於swig_import_helper _mod = imp.load_module(' _pywrap_tensorfl ow_internal',fp,路徑名,描述)文件“ /usr/lib/python3.6/imp.py”,第243行,在load_module中返回load_dynamic(名稱,文件名,文件)文件“ /usr/lib/python3.6/ imp.py“,行343,在load_dynamic中返回_load(spec)ImportError:libcublas.so.9.0:無法打開共享庫文件:沒有這樣的文件或目錄

無法加載本地TensorFlow運行時。

請參閱https://www.tensorflow.org/install/install_sources#common_installation_problems

出於某些常見原因和解決方案。 尋求幫助時,在此錯誤消息上方包括整個堆棧跟蹤。

我已經嘗試使用ubuntu 16.04,但是GPU不支持。 也不支持安裝CUDA9.0&CUDA9.2。

我如何使用tensorflow-gpu?

我已經在〜/ .bashrc中應用了路徑

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64$LD_LIBRARY_PATH

tensorflow-gpu軟件包是針對Cuda 9.0構建的,但是您已經安裝了Cuda 10.0。

您需要將Cuda的版本降級到9.0(但如果我記得,使用2080Ti是不可能的),或者從源代碼構建tensorflow。 tensorflow Webiste上有大量有關如何執行此操作的文檔

您也可以嘗試安裝軟件包tf-nightly-gpu 您應該注意,該版本是更具實驗性的,因為尚未經過廣泛的測試。

暫無
暫無

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

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