我已经安装了redhat 6.6,cuda 9.0,cudnn 7.3.1和tesnorflow-gpu。但是安装后出现以下错误。我的显卡是nvidia titan v 追溯(最近一次通话):
<Label Text="Sample &Text"/>
答案 0 :(得分:1)
您是否测试了cuda和cudnn的安装:
Nvidia驱动程序:
$ nvidia-smi
这应该向您显示GPU指标的概述
测试CUDA:
请先检查是否:
$ nvcc -V
显示正确的cuda工具包版本
然后您可以通过以下过程对其进行测试: 首先:
$ cd ~/NVIDIA_CUDA-9.0_Samples
$ make
然后:
$ cd ~/NVIDIA_CUDA-10.0_Samples/bin/x86_64/linux/release
$./deviceQuery
如果最后有一个“结果:通过”,那就一切都很好!
要测试cudnn:
$ cp -r /usr/src/cudnn_samples_v7/ $HOME
$ cd $HOME/cudnn_samples_v7/mnistCUDNN
$ make clean && make
$ ./mnistCUDNN
结果应该是:“测试通过!”
tensorflow:
如果cuda和cudnn在工作,我建议您使用以下命令在conda环境中安装tensorflow:
conda create --name tf_gpu tensorflow-gpu
对我来说(并且在遇到很多问题之后),它运行得很好。
要测试:
from tensorflow.python.client import device_lib
device_lib.list_local_devices()