conda 安装 tensorflow-gpu

一. 背景

今天尝试在conda创建的虚拟环境里安装tensorflow,使用如下代码进行安装

pip install tensorflow

但是会报错,如下:

$ pip install tensorflow
Downloading/unpacking tensorflow
  Could not find any downloads that satisfy the requirement tensorflow
Cleaning up...
No distributions at all found for tensorflow

二. 解决方案

后来找了很多资料,发现可以直接用Anconda直接安装,并且实现并不需要预先知道tensorflow的路径。

1. 加载虚拟环境

source activate venv

2. 使用conda查询可安装包

(venv)$ anaconda search -t conda tensorflow-gpu
Using Anaconda API: https://api.anaconda.org
Packages:
     Name                      |  Version | Package Types   | Platforms       | 
     ------------------------- |   ------ | --------------- | --------------- | 
     HCC/tensorflow-gpu        |    1.7.0 | conda           | linux-64        | 
     aaronzs/tensorflow-gpu    |   1.10.0 | conda           | linux-64, win-64 | 
     acaprez/tensorflow-gpu    |    1.5.0 | conda           | linux-64        | 
.
.
.
.(这里其实还有很多,就不贴出来了)
machine learning.
variant.
Found 23 packages

Run 'anaconda show <USER/PACKAGE>' to get installation details

3. 使用anaconda查看安装路径

anaconda show aaronzs/tensorflow-gpu

Using Anaconda API: https://api.anaconda.org
Name:    tensorflow-gpu
Summary: TensorFlow helps the tensors flow
Access:  public
Package Types:  conda
Versions:
   + 1.3.0
   + 1.4.0rc1
   + 1.4.0
   + 1.5.0
   + 1.6.0
   + 1.7.0
   + 1.8.0
   + 1.7.1
   + 1.9.0
   + 1.10.0

To install this package with conda run:
     conda install --channel https://conda.anaconda.org/aaronzs tensorflow-gpu

4. 进行安装

conda install --channel https://conda.anaconda.org/aaronzs tensorflow-gpu=1.10.0

三. 参考资料

1. anaconda安装tensorflow

### 如何在 Conda 环境下成功安装 TensorFlow #### 创建并激活新的 Conda 环境 为了确保兼容性和减少冲突,建议创建一个新的 Conda 环境来专门用于 TensorFlow安装。可以按照以下方式操作: ```bash conda create -n tensorflow_env python=3.7 ``` 此命令会创建名为 `tensorflow_env` 的新环境,并指定 Python 版本为 3.7[^3]。 激活该环境的方法如下: ```bash conda activate tensorflow_env ``` #### 安装 TensorFlow 一旦环境被激活,在其中可以直接通过 Conda 或 pip 来安装 TensorFlow。以下是两种方法的具体实现: ##### 方法一:使用 Conda 安装 Conda 提供了一个简单的方式来管理包及其依赖项。可以通过以下命令安装最新版的 TensorFlow: ```bash conda install -c anaconda tensorflow ``` 如果需要特定版本(例如 TensorFlow 2.2),则可执行以下命令: ```bash conda install tensorflow==2.2 ``` 注意,有时由于通道不可访问或其他原因可能会遇到错误消息,如 UnavailableInvalidChannel 错误[^2]。此时应尝试更新 Conda 并清理缓存后再重试: ```bash conda update conda conda clean --all ``` ##### 方法二:使用 Pip 安装 另一种常见的方式是利用 pip 工具完成 TensorFlow安装。首先确认 pip 是否已正确安装于环境中: ```bash pip --version ``` 随后即可运行以下指令以安装所需版本的 TensorFlow: 对于仅支持 CPU 的情况: ```bash pip install tensorflow==2.2 ``` 而对于 GPU 支持的情况,则需安装对应的 GPU 加速版本: ```bash pip install tensorflow-gpu==2.1.0 ``` 需要注意的是,GPU 版本通常还需要额外配置 NVIDIA 显卡驱动程序以及 CUDA 和 cuDNN 库的支持[^5]。 #### 验证安装是否成功 最后一步是在 Python 中验证 TensorFlow 是否正常工作。打开终端或者脚本文件输入以下代码测试其功能: ```python import tensorflow as tf print(tf.__version__) ``` 当输出显示为期望的版本号时即表明安装无误[^1]。 #### 常见问题处理 - 如果出现关于通道无法访问的问题,请参照上述提到的解决方案进行排查。 - 对于 GPU 用户来说,除了基本软件外还需关注硬件条件满足度,比如显卡型号、CUDA/cuDNN 版本匹配等问题[^4]。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值