如何安装cuda

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了如何安装cuda相关的知识,希望对你有一定的参考价值。

参考技术A 安装cuda时,第一次会让设置临时解压目录,第二次会让设置安装目录;

临时解压路径,建议默认即可,也可以自定义。安装结束后,临时解压文件夹会自动删除;

安装目录,建议默认即可;

注意:临时解压目录千万不要和cuda的安装路径设置成一样的,否则安装结束,会找不到安装目录的!!!

选择自定义安装

安装完成后,配置cuda的环境变量;

命令行中,测试是否安装成功;

cuda8.0如何安装TensorFlow-gpu?

【中文标题】cuda8.0如何安装TensorFlow-gpu?【英文标题】:How to install TensorFlow-gpu with cuda8.0? 【发布时间】:2018-07-12 13:18:18 【问题描述】:

我尝试按照官网的说明安装,导致导入tensorflow时出现ImportError:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

我运行代码cat /usr/local/cuda/version.txt,显示我的cuda版本是8.0.61。

似乎 tensorflow 正在寻找 cuda 9.0。我无法升级 cuda,因为我在共享 gpu-server 上工作并且我没有 root 权限。

有没有办法让 tensorflow 与 cuda 8.0 一起工作?或者有什么其他的方法?

谢谢!!

【问题讨论】:

【参考方案1】:

您需要为 CUDA-8 安装 1.4.1 版本

pip install tensorflow-gpu==1.4.1

最新的(版本1.5)适用于CUDA-9

【讨论】:

为什么我安装 1.4.1,它显示错误:ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory。它是否说它需要cuda 6.5?如何使用 CUDA8.0? @sam 导入错误 (ImportError: libcudnn.so.6) 表示您需要适当的 CuDNN 库 (developer.nvidia.com/cudnn)。 CuDNN 与 CUDA 是分开的。正如答案所示,tensorflow-gpu==1.4.1 适用于 CUDA 8。安装 CuDNN 的说明可以在讨论中找到:askubuntu.com/questions/767269/… 谢谢,您从哪里得到这些信息? github.com/tensorflow/tensorflow/releases/tag/v1.5.0 - 在 v1.5.0 的发行说明中,作者提到二进制文件现在是针对 CUDA 9 进行预构建的【参考方案2】:

我遇到了类似的问题,直到我发现

https://www.tensorflow.org/install/install_sources#tested_source_configurations

检查您安装的 cuda 版本和 cudnn 版本,然后找出哪个版本的 tensorflow-gpu 与使用上述链接的版本兼容。

我已经安装了 cuda 8 和 cudnn v5.1,因此通过检查上面的链接 tensorflow-gpu 1.2.0 是兼容的,并且在安装之后使用

pip install tensorflow-gpu==1.2.0

它对我有用。

【讨论】:

以上是关于如何安装cuda的主要内容,如果未能解决你的问题,请参考以下文章

如何获得 CUDA 版本?

cuda 如何安装 18.04 ubuntu

如何在未安装 CUDA 工具包的机器上运行已编译的 CUDA 代码?

cuda8.0如何安装TensorFlow-gpu?

如何在 nvidia/cuda 基础 docker 中仅安装 nvcc?

如何在NVIIA Jeston K1上安装CUDA环境