在cuda 9和cudnnv7上安装tensorflow-gpu 1.4

Question

我试图在Windows 10上使用cuda 9和cudnn v7在gtx 1050上安装tensorflow-gpu。我添加了所有环境变量。但是我收到以下错误：ImportError：找不到'cudart64_80.dll'。 TensorFlow要求将此DLL安装在％PATH％环境变量中指定的目录中

Answer 1

我只是回复了一些非常相关的in this other SO post。假设您已经获得了CUDA和CUDNN部分，并且对于您的具体问题，在TensorFlow 1.4上使用CUDA9 + CUDNN7在Python上确实可行且容易：

1. 访问mind's repo并下载适合您需求的wheel文件。 wheel文件是预编译的Python软件包，可以很容易地用pip安装，希望它也可以在Windows上运行而不会有太多麻烦（我没有测试它，让我知道这是否有效）。如果你想要的包不在那里，你可以看看用户this repo组织的Yaroslav Bulatov，看看你是否找到了你想要的东西。顺便说一下，谢谢大家的努力！
2. 用pip安装车轮。快速测试：

   import tensorflow as tf
   x = tf.nn.conv2d(tf.ones([1,1,10,1]), tf.ones([1,5,1,1]), strides=[1, 1, 1, 1], padding='SAME')
   with tf.Session() as sess:
       sess.run(x) # this should output a tensor of shape (1,1,10,1) with [3,4,5,5,5,5,5,5,4,3]

请注意，正如here所说，一些心灵车轮需要安装英特尔的MKL库。同样，在我之前提到的另一篇SO帖子中，我详细介绍了安装它并让TF在Linux上找到它的说明（Linux用户可能想查看我链接的帖子，了解有关安装没有root权限的CUDA和CUDNN的更多详细信息）。建立它的官方说明可以找到here（它们似乎只适用于Ubuntu）。

干杯， 安德烈斯

Answer 2

当前版本的张量流即1.4支持cuda 8，因此缺少cuda 8的'cudart64_80.dll'。下载cuda 8和合适的cudnn，它将解决问题