作者:禅与计算机程序设计艺术
1.简介
TensorFlow是一个开源的深度学习框架,它为机器学习提供了一种新的思维方式。本文将从零开始介绍TensorFlow并基于Google Colab进行使用示例,带领读者走进TensorFlow的世界。
2.安装配置
安装TF
conda install tensorflow
如果你的计算机上没有GPU支持,或者想用CPU版,则可以只安装CPU版本:
pip install tensorflow-cpu
Linux和Mac系统上可以使用以下命令进行安装(可能需要添加sudo):
pip install tensorflow
配置GPU
如果你的电脑有Nvidia显卡,并且已经安装CUDA、CuDNN并成功启动,那么你可以通过以下方法配置GPU环境:
- 安装tensorflow-gpu版本:
!pip uninstall