最近又在鼓捣tensorflow,搞到一个比较简单的安装方法,在这里分享一下
之前在网上看到的tensorflow安装教程,尤其是GPU版因为依赖cuda和cudnn,所以很多教程都是分开单独安装,很多小伙伴搞不清楚一不小心就会有版本冲突问题,这里介绍一个简单的方法
1. 安装anaconda,安装就不过多赘述,按各自需要安装就好
anaconda官网下载安装
2. anaconda配置
打开cmd切换到国内的镜像源
- 切换成清华的镜像
conda config --add channels /anaconda/pkgs/free/
conda config --add channels /anaconda/pkgs/main/
-
切换成中科大的镜像
conda config --add channels /anaconda/pkgs/free/
conda config --add channels /anaconda/pkgs/main/
注意!如果切换镜像后当出现下载不了的情况,就先切换默认源,然后再修改另一个可以使用的conda源(一定要先恢复默认,再换另一个!!!)
- 切回默认源
conda config --remove-key channels
其他镜像网站可以自行上网搜索
3. TensorFlow安装
这里建议安装tensorflow 1.15的版本,需要安装其他版本的只需要在安装tensorflow的命令行中修改对应版本号即可。
打开cmd运行,首先创造tensorflow 1.15需要的环境
conda create -n tensorflow pip python=3.6
激活TensorFlow环境
activate tensorflow
用conda安装tensorflow
CPU版输入 conda install tensorflow=1.15.0
GPU版输入 conda install tensorflow-gpu=1.15.0
(重点!)选择conda安装而不用pip安装,是因为在安装tensorFlow-gpu版的过程中,它会自动配置对应版本号的cuda和cudnn,而不需要再单独安装
(这里截图是tensorflow1.12的安装过程,因为我的电脑只能支持cuda9.0,安装tensorflow1.15的过程也是这样,只是cuda版本号会不一样而已)
然后等待安装完成就好了,不能装gpu版的就老老实实装cpu版!!!
之前作为第一次安装tensorflow的人也踩了不少坑,这算是我目前看到的最简单也最不容易出错的一个安装方法吧,如果有小伙伴还没装过,或者安装出错解决不了的都可以试一试
happy coding~