Python 无法在DGX-2特斯拉上安装带有CUDA和Cudnn的Tensorflow
我试图在特斯拉GPU上运行我的tf代码。但是当我在nvcr.io/nvidia/tensorflow:19.01-py3_v3图像上创建一个pod时。但是,在导入tensorflow时,我得到了以下堆栈跟踪。任何帮助都将不胜感激。谢谢Python 无法在DGX-2特斯拉上安装带有CUDA和Cudnn的Tensorflow,python,tensorflow,Python,Tensorflow,我试图在特斯拉GPU上运行我的tf代码。但是当我在nvcr.io/nvidia/tensorflow:19.01-py3_v3图像上创建一个pod时。但是,在导入tensorflow时,我得到了以下堆栈跟踪。任何帮助都将不胜感激。谢谢 Traceback (most recent call last): File "/home/srivastava6/.local/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow.
Traceback (most recent call last):
File "/home/srivastava6/.local/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "/home/srivastava6/.local/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "/home/srivastava6/.local/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
File "/usr/lib/python3.5/imp.py", line 242, in load_module
return load_dynamic(name, filename, file)
File "/usr/lib/python3.5/imp.py", line 342, in load_dynamic
return _load(spec)
ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory
回溯(最近一次呼叫最后一次):
文件“/home/srivastava6/.local/lib/python3.5/site packages/tensorflow/python/pywrap\u tensorflow.py”,第58行,在
从tensorflow.python.pywrap\u tensorflow\u内部导入*
文件“/home/srivastava6/.local/lib/python3.5/site packages/tensorflow/python/pywrap\u tensorflow\u internal.py”,第28行,in
_pywrap\u tensorflow\u internal=swig\u import\u helper()
swig\u import\u helper中的文件“/home/srivastava6/.local/lib/python3.5/site packages/tensorflow/python/pywrap\u tensorflow\u internal.py”,第24行
_mod=imp.load_模块(“pywrap_tensorflow_internal”,fp,路径名,描述)
加载模块中的文件“/usr/lib/python3.5/imp.py”,第242行
返回加载动态(名称、文件名、文件)
文件“/usr/lib/python3.5/imp.py”,第342行,动态加载
返回负载(规格)
ImportError:libcublas.so.8.0:无法打开共享对象文件:没有此类文件或目录
CUDA似乎有问题,您有什么版本的CUDA?如果您自己降级或编译tensorflow的版本是10.1,那么问题是我们使用的代码需要TF 1.3/1.4/1.5,而TF 1.3/1.4/1.5与tensorflow gpu不兼容,因此会出现错误