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

tensorflow -gpu安装方法(不用自己装cuda,cdnn)

程序员文章站 2023-01-05 18:15:58
tensorflow™是一个基于数据流编程(dataflow programming)的符号数学系统,被广泛应用于各类机器学习(machine learning)算法的编程...

tensorflow™是一个基于数据流编程(dataflow programming)的符号数学系统,被广泛应用于各类机器学习(machine learning)算法的编程实现,其前身是谷歌的神经网络算法库distbelief [1]  。

tensorflow拥有多层级结构,可部署于各类服务器、pc终端和网页并支持gpu和tpu高性能数值计算,被广泛应用于谷歌内部的产品开发和各领域的科学研究 [1-2]  。

tensorflow -gpu安装

首先,安装anoconda

1. 官网下载点我:

2.安装

点击 python 3.6 version自动下载x64版,下载好之后,然后安装。

如图,打上勾之后,一路next

3.打开终端

1)输入conda –version 查看版本

2)配置python环境

我装的是python3.5,你们可以根据情况自己选择

conda create –n tensorflow python=3.5

3)激活python环境:activate tensorflow

你可以退出当前环境:deactivate tensorflow

划重点了:安装tensorflow

**传统的方法就是用pip 安装,导致后面还要下载英伟达的cuda支持以及cdnn库**
**我在这里介绍一种简单的方法,不依靠pip**

在终端中输入

conda install tensorflow-gpu

2.看包中内容

如图,cuda 9.0和cdnn 7.1.4已经自动安装了不用再自己下载

3.安装完之后

打开anaconda prompt 测试成功

总结

以上所述是小编给大家介绍的tensorflow -gpu安装 ,最简单的途径(不用自己装cuda,cdnn),希望对大家有所帮助