[英]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.