Tensorflow:我安装了CUDA 9.2,但它需要9.0吗?

13

我按照书中的指示安装了CUDA Toolkit 9.2版本,但是遇到了这个错误:

ImportError: 找不到 'cudart64_90.dll'。TensorFlow要求将此DLL安装在你的%PATH%环境变量中命名的目录下。从此链接下载并安装CUDA 9.0:https://developer.nvidia.com/cuda-toolkit

我搜索并发现tensorflow需要9.0版本,但是接下来该怎么做呢?我应该删除旧版本的CUDA然后安装9.0版本,还是仅安装9.0版本(这样我就有两个版本共存了)?我不想弄坏我的电脑,请帮忙~~~

编辑:

我从这个链接下载了CUDA9.0补丁1:https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal。但好像安装程序没有运行(我双击它后立即消失了)?


补丁版本不是完整的安装程序。正如名称所示,它是现有CUDA 9安装的补丁。下次阅读下载页面上的内容。 - talonmies
@talonmies:感谢您的提醒,但实际上CUDA9.0的patch2是一个完整的安装程序(我的意思是它的界面与安装CUDA9.0相同),而patch1只是一个“闪烁”(就像命令提示符的结果没有系统暂停)。这就是为什么我认为这可能是一个错误的原因。 - VimNing
1
不是这样的。请重新阅读下载页面上的内容--“基础安装程序可在下面下载。 有2个补丁可用。这些补丁需要先安装基础安装程序。”(强调我的) - talonmies
Google Cloud现在提供了CUDA 9.2的TensorFlow 1.10镜像。您可以创建一个带有GPU的虚拟机,并选择其中一个预安装的TensorFlow镜像。创建的虚拟机还将具有不同Python版本的.whl文件。您可能能够从虚拟机中复制.whl文件并将其安装到其他地方,但我尚未尝试过。 - scv
5个回答

14

谢谢,这很有用。对于那些想要使用.run文件进行累加安装cuda9.0的人,请在运行安装程序时不要安装驱动程序(当被询问时选择“否”)。我不知道为什么NVIDIA在安装指南中没有提供这个信息。 - Chan Kim

1

我曾经遇到同样的问题,后来通过将Python 3.5更改为Python 3.6解决了它。希望你也能解决它。


刚试了一下,不起作用(Windows7,tensorflow-1.9 二进制安装)。 - mieszkoz

0

请按照以下链接中提到的所有步骤进行操作。我已经成功在Ubuntu 18.04 上安装了Tensorflow。

http://www.python36.com/how-to-install-tensorflow-gpu-with-cuda-9-2-for-python-on-ubuntu/

在确认安装了NVIDIA驱动程序之后,您需要安装cuDNN、NCCL和依赖项(libcupti-dev和添加LD_LIBRARY_PATH)。然后从源代码配置TensorFlow。

wget https://github.com/bazelbuild/bazel/releases/download/0.14.0/bazel-0.14.0-installer-linux-x86_64.sh
chmod +x bazel-0.14.0-installer-linux-x86_64.sh
./bazel-0.14.0-installer-linux-x86_64.sh --user
echo 'export PATH="$PATH:$HOME/bin"' >> ~/.bashrc

并重新加载环境变量。通过下载最新的tensorflow开始构建TensorFlow进程,随后您需要使用bazel构建Tensorflow。此过程需要很长时间,可能需要数小时。

git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
git pull
git checkout r1.8
./configure
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

构建 whl 文件,请执行以下命令:
bazel-bin/tensorflow/tools/pip_package/build_pip_package tensorflow_pkg

使用pip安装tensorflow:

cd tensorflow_pkg

对于已有的虚拟环境:

pip install tensorflow*.whl

之后,您可以通过运行一个简单的示例来验证您的TensorFlow安装。


0
如果您安装了更高版本的Cuda,请运行较低版本的Cuda安装程序,它将自动删除已安装的版本(如果不是相同的版本)。在我的情况下,在Windows 10上,Cuda9.2被Cuda9.0安装程序删除。谢谢!

0

我已经在我的Ubuntu 16.4上安装了cuda-9.2,并尝试仅安装cuda 9.0库。但是,sudo apt-get install cuda-libraries-9-0无法工作,给出以下消息:E:无法定位软件包cuda-libraries-9-0。 - Chan Kim

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接