在Linux服务器上配置tensorflow-gpu版(最详细教程)

       由于基于tensorflow的神经网络在CPU上运行速度太慢,在GPU上运行速度会快很多,但是在linux系统上tensorflow-GPU运行配置比较麻烦,有很多坑需要踩,网上的很多教程和自己使用的环境不匹配,所以针对这个情况自己做一个总结,总结一下tensorflow-gpu版本在linux服务器上的配置过程,及其中遇到的问题,给大家提供参考和交流。

环境:linux+Anaconda3+python3.5+CUDA9.0+cudnn7.0

安装流程:

一. Anaconda下载及安装

二. 建立tensorflow gpu版本环境(包括配置清华镜像)

三. CUDA下载及安装

四. cudnn下载及配置

五. pycharm下载及安装

一. Anaconda下载及安装
官网地址:https://www.anaconda.com/distribution/#download-section

清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

下载linux系统的Python3.7 version。

输入命令bash bash Anaconda3-2019.03-Linux-x86_64.sh安装Anaconda。

显示

在标红框处输入yes和安装目标路径,

继续选择yes,

等待安装完成。安装完成后,输入conda --version如果显示版本信息则安装成功。

如果如下图所示无法显示

输入source ~/.bashrc,

Anaconda安装完成。

二. 建立tensorf

### 安装 TensorFlow-GPU 的方法 为了在 Linux 服务器上使用 Anaconda 正确配置安装 TensorFlow-GPU,可以通过以下几种方式进行操作: #### 方法一:通过 Conda 命令直接安装 可以直接利用 `conda` 来安装指定本的 TensorFlow-GPU。例如,要安装 TensorFlow-GPU 本 1.12.0,可以运行以下命令: ```bash conda install tensorflow-gpu==1.12.0 ``` 此方法适用于 Conda 已经支持的 TensorFlow-GPU 本[^1]。 #### 方法二:通过 Pip 安装特定本 如果目标 TensorFlow-GPU 本未被 Conda 支持,则可以在激活的 Anaconda 虚拟环境中使用 `pip` 进行安装。例如,对于 TensorFlow-GPU 1.13.2 可以执行以下命令: ```bash pip install tensorflow-gpu==1.13.2 ``` 这种方法允许更灵活地选择不同本的 TensorFlow-GPU[^2]。 #### 方法三:离线安装 TensorFlow-GPU 当网络条件受限时,可以选择下载对应本的 `.whl` 文件并通过本地文件路径完成安装。例如,在 Python 3.7 环境下安装 TensorFlow-GPU 2.3.0,可先获取对应的轮子文件(wheel file),然后运行以下命令: ```bash pip install tensorflow_gpu-2.3.0-cp37-cp37m-manylinux2010_x86_64.whl ``` 该过程需确保所选 wheel 文件与系统的架构以及 Python 解释器本相匹配[^3]。 #### 额外提示:混合工具链安装其他依赖项 有时可能还需要额外安装一些库来满足项目需求。在这种情况下,即使某些包无法通过 Conda 获取,也可以借助内置的 Pip 实现安装目的。比如安装 NumPy 库至当前活动的虚拟环境里,可以用下面这条指令实现: ```bash python -m pip install numpy ``` 这一步骤展示了如何结合两种工具的优势来构建完整的开发环境[^4]。
评论 13
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值