记录一次成功配置anaconda + tensorflow-gpu环境
本文主要提供一个成功的anaconda+tensorflow-gpu环境的各模块版本示例。
版本
- CUDA 9.2
- cudnn 7.4
- python 3.6
- tensorflow-gpu 1.10.0
- numpy 1.16(版本过高,会有warning)
命令
CUDA和cudnn是师弟已经装好的,这部分只要注意版本适配问题就可以。tensorflow-gpu的安装试了很多组版本,失败了很多次,装了一天半终于弄好了。
- 创建名为tensorflow 的环境并激活,Python版本为3.6
conda create -n tensorflow python=3.6
conda activate tensorflow - 安装tensorflow-gpu 1.10.0
conda install tensorflow-gpu==1.10.0 - 安装numpy 1.16
conda install numpy==1.16
这一步是因为numpy 版本过高,在import tensorflow 时会出现警告。