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在Python上使用CUDA9 + CUDNN7确实可行且容易：

访问mind's repo并下载符合您需求的wheel文件。 Wheel文件是预编译的Python软件包，可以使用pip轻松安装，希望它也可以在Windows上运行而不会有太多麻烦（我没有测试它，请告诉我这是否有效）。如果您所需的套餐不在那里，您可以查看用户this repo组织的Yaroslav Bulatov，看看您是否找到了您要查找的内容。顺便说一下，谢谢大家的努力！

使用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]

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

干杯，
安德烈

Answer 2

当前版本的tensorflow，即1.4支持cuda 8，因此＆＃39; cudart64_80.dll＆＃39; cuda 8失踪了。下载cuda 8和合适的cudnn，它将解决问题

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

2 个答案: