如何安装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 代码?