简要说明一下Win10系统下搭建Tensorflow环境方法。
1.首先Anaconda下载与搭建
(1)下载
官网地址 https://www.anaconda.com/distribution/
最新版本的python3.7版本,直接下载即可,速度很慢,推荐使用镜像下载。
镜像推荐使用清华镜像下载,因为清华大学已经拥有Anaconda的合法镜像使用权(感谢清华大学-v0v-)
清华镜像地址 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
各种版本都有,并且经常更新。
下载完成后打开。
(2)搭建环境
点到Environment
然后Create一个新环境
名字任意Python版本选择3.7。
然后搜索框选择All,搜索tensorflow
有CPU和GPU版本,一般深度学习采有GPU版本,GPU版本需要有显卡支持。Anaconda里的gpu版本的包集成了CUDA和CUDNN,所以必须要单独下载(这点需要大赞(#.#)),但是需要将显卡驱动更新到最新来适配最新的CUDA和CUDNN,如果需要CPU版本直接勾选下载即可,不需要其他操作。
如果采用官方的Channel下载速度会很慢甚至出错,还是推荐使用清华镜像的Channel
下面讲解一下换源方法,将官方源换成清华镜像。
首先打开Anaconda里的命令窗口Prompt
conda config --show-sources
如下图,显示两个Channels,一个defult是官方源,另一个就是清华镜像。
导入清华镜像需要下面命令
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
导入之后用之前的命令检测是否导入成功。
然后在软件界面Channels选项查看·
同时可以移除自己不需要的源。另外Anaconda一般下载的各种pkgs版本都是最新的,可以根据自己的需要升级和降级版本很方便。点击左边的绿色对勾即可出现下拉菜单可以选择版本。
搭建好之后Anaconda部分就结束了。
Pycharm的安装和适配
Pycharm的下载
官网地址 https://www.jetbrains.com/pycharm/download/#section=windows
下載安裝即可。
打开Pycharm - File - setting - Project-Project Interpreter
进入点击右上角设置选择add
选择Existing enviroment,按照图上所示路径(根据自己Anaconda安装路径调整)关联到Anaconda
然后Pycharm下载和搭建部分完成
接下来进行代码测试
Tensorflow官网的手写数字识别
import tensorflow as tf
mnist = tf.keras.datasets.mnist
(x_train, y_train),(x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(10, activation='softmax')
])
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
model.fit(x_train, y_train, epochs=5)
model.evaluate(x_test, y_test)
在Pycharm中导入
结果如图
测试成功,全部搭建完毕。