欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2024-03-18 13:40:10
...

最近又在鼓捣tensorflow,搞到一个比较简单的安装方法,在这里分享一下

之前在网上看到的tensorflow安装教程,尤其是GPU版因为依赖cuda和cudnn,所以很多教程都是分开单独安装,很多小伙伴搞不清楚一不小心就会有版本冲突问题,这里介绍一个简单的方法

1.  安装anaconda,安装就不过多赘述,按各自需要安装就好

anaconda官网下载安装

2.  anaconda配置

打开cmd切换到国内的镜像源(为什么要切换原因不用我说了吧...)

  • 切换成清华的镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  • 切换成中科大的镜像 

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/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,而不需要再单独安装

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

(这里截图是tensorflow1.12的安装过程,因为我的电脑只能支持cuda9.0,安装tensorflow1.15的过程也是这样,只是cuda版本号会不一样而已)

然后等待安装完成就好了,不能装gpu版的就老老实实装cpu版!!!

之前作为第一次安装tensorflow的人也踩了不少坑,这算是我目前看到的最简单也最不容易出错的一个安装方法吧,如果有小伙伴还没装过,或者安装出错解决不了的都可以试一试

happy coding~

 

相关标签: anaconda tensorflow