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())
结果为True 说明已经安装成功。