search

NVIDIA-AI-IOT / tf_trt_models

TensorFlow models accelerated with NVIDIA TensorRT
BSD 3-Clause "New" or "Revised" License
681 stars 245 forks source link

cuda version problem causes cannot use tensorflow #54

Open i-chaochen opened 5 years ago

i-chaochen commented 5 years ago

I think the latest jetson tx2 is cuda 10.0, but the tutorial install the tensorflow with cuda9.0, so when I import tensorflow, I met the following error:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

In [1]: import tensorflow as tf

ImportError Traceback (most recent call last)

in () ----> 1 import tensorflow as tf /home/nvidia/.local/lib/python2.7/site-packages/tensorflow/__init__.py in () 20 21 # pylint: disable=g-bad-import-order ---> 22 from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import 23 24 try: /home/nvidia/.local/lib/python2.7/site-packages/tensorflow/python/__init__.py in () 47 import numpy as np 48 ---> 49 from tensorflow.python import pywrap_tensorflow 50 51 # Protocol buffers /home/nvidia/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py in () 72 for some common reasons and solutions. Include the entire stack trace 73 above this error message when asking for help.""" % traceback.format_exc() ---> 74 raise ImportError(msg) 75 76 # pylint: enable=wildcard-import,g-import-not-at-top,unused-import,line-too-long ImportError: Traceback (most recent call last): File "/home/nvidia/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in from tensorflow.python.pywrap_tensorflow_internal import * File "/home/nvidia/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in _pywrap_tensorflow_internal = swig_import_helper() File "/home/nvidia/.local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description) ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory