Tensorflow安装教程 傻瓜式一键安装

时间:2024-10-26 08:35:14

最近又在鼓捣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~