Conda 安装 tensorflow gpu 1.13.1(验证可行)

Conda 安装 tensorflow gpu 1.13.1

conda 有虚拟环境管理的功能,本次我来介绍在一个 conda 环境中配置tensorflow gpu 版本

conda create -n tft python==3.6 我这里的环境命名为 tft

conda activate tft 切换到 tft 环境中进行操作

conda install cudatoolkit=10.0

conda install tensorflow-gpu=1.13.1

pip install tensorflow-gpu==1.13.1

如果 pip 速度太慢,使用 (总是会出现)

pip install tensorflow-gpu==1.13.1 -i https://pypi.douban.com/simple --trusted-host=pypi.douban.com

如果出现 numpy 版本太高的警告(总是会出现)

pip uninstall numpy
pip install numpy==1.16.4

记得加 -i 进行加速,马上飞起。

测试

import tensorflow as tf
print(tf.test.is_gpu_available())

Conda 安装 tensorflow gpu 1.13.1(验证可行)_第1张图片
结果为True 说明已经安装成功。

你可能感兴趣的:(TensorFlow,anaconda,tensorflow)