2017-02-23 82 views
0

的最新版本的Tensorflow(1.0)中安装CUDA 8.0我已经升级到Tensorflow 1.0版,并安装了CUDA 8.0 cudnn 5.1版本和nvidia驱动程序最新375.39。我的NVIDIA硬件是Amazon Web Services上使用p2.xlarge实例(一种特斯拉K-80)的硬件。我的操作系统是Linux 64位。如何在AWS p2.xlarge实例,AMI ami-edb11e8d和nvidia驱动程序最新版本(375.39)

每次都遇到我用命令的时间一个错误信息:tf.Session()

[[email protected] CUDA]$ python 
Python 2.7.12 (default, Sep 1 2016, 22:14:00) 
[GCC 4.8.3 20140911 (Red Hat 4.8.3-9)] on linux2 
Type "help", "copyright", "credits" or "license" for more information. 
>>> import tensorflow as tf 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcublas.so.8.0 locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcudnn.so.5 locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcufft.so.8.0 locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcuda.so.1 locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcurand.so.8.0 locally 
>>> sess = tf.Session() 
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE3 instructions, but these are available on your machine and could speed up CPU computations. 
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations. 
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations. 
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations. 
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX2 instructions, but these are available on your machine and could speed up CPU computations. 
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use FMA instructions, but these are available on your machine and could speed up CPU computations. 
E tensorflow/stream_executor/cuda/cuda_driver.cc:509] failed call to cuInit: CUDA_ERROR_NO_DEVICE 
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:158] retrieving CUDA diagnostic information for host: ip-172-31-7-96 
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:165] hostname: ip-172-31-7-96 
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:189] libcuda reported version is: Invalid argument: expected %d.%d or %d.%d.%d form for driver version; got "1" 
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:363] driver version file contents: """NVRM version: NVIDIA UNIX x86_64 Kernel Module 375.39 Tue Jan 31 20:47:00 PST 2017 
GCC version: gcc version 4.8.3 20140911 (Red Hat 4.8.3-9) (GCC) 
""" 
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:193] kernel reported version is: 375.39.0 

我完全茫然不知如何解决这个问题。 我已经尝试过不同版本的Nvidia驱动程序和CUDA,但仍然无法正常工作。

任何提示将不胜感激。

+1

也许是你的GPU驱动程序安装不正确。运行'nvidia-smi'的结果是什么?您是否按照[cuda linux安装指南](http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#abstract)中所述执行了CUDA安装的任何验证? –

+0

谢谢您的及时答复。 nvidia-smi工作,我没有按照网站上描述的“验证”。 我决定在Redhat 7.3系统上从头开始。 它起初工作,所以不需要进一步的援助。 – basuam

回答

0

卸载驱动程序& cuda,然后按照official guide重新安装。

运行deviceQuery来检查设备是否安装正确。

+0

非常感谢您的回复。正如你所建议的那样,我从头开始安装所有东西后运行了deviceQuery。我使用RedHat 7.3创建了另一个实例,并花费了一些时间更新所有软件包。最后,它工作得很好。 – basuam

+0

很高兴现在正在工作! – tblue

2

您需要安装NVIDIA驱动程序并运行CUDA 8.0安装程序。

# Requirements 
# - NVIDIA Driver - NVIDIA-Linux-x86_64-375.39.run - http://www.nvidia.fr/Download/index.aspx 
# - CUDA runfile (local) - cuda_8.0.61_375.26_linux.run - https://developer.nvidia.com/cuda-downloads 
# - cudnn-8.0-linux-x64-v5.0-ga.tgz 

sudo apt update -y && sudo apt upgrade -y 
sudo apt install build-essential linux-image-extra-`uname -r` -y 

chmod +x NVIDIA-Linux-x86_64-375.39.run 
sudo ./NVIDIA-Linux-x86_64-375.39.run 

chmod +x cuda_8.0.61_375.26_linux.run 
./cuda_8.0.61_375.26_linux.run --extract=`pwd`/extracts 
sudo ./extracts/cuda-linux64-rel-8.0.61-21551265.run 

echo -e "export CUDA_HOME=/usr/local/cuda\nexport PATH=\$PATH:\$CUDA_HOME/bin\nexport LD_LIBRARY_PATH=\$LD_LINKER_PATH:\$CUDA_HOME/lib64" >> ~/.bashrc 
source .bashrc 

tar xf cudnn-8.0-linux-x64-v5.0-ga.tgz 
cd cuda 
sudo cp lib64/* /usr/local/cuda/lib64/ 
sudo cp include/cudnn.h /usr/local/cuda/include/ 
+0

我以为cudnn 8不支持张量流 – Goddard

相关问题