一、环境搭建
1、安装pycharm3.6.4及对应版本的Anaconda
安装包百度云链接:
百度云链接 提取码:m7xa
2、安装Tensorflow
Tensorflow有CPU版本与GPU版本
通过pip命令安装CPU版本: pip install tensorflow
通过pip命令安装GPU版本: pip install tensorflow-gpu
如果速度太慢,可使用清华镜像:
pip install (包名称) -i https://pypi.tuna.tsinghua.edu.cn/simple
测试tensorflow:
打开命令行,输入python进入python环境
在输入
import tensorflow as tf
不报错即可,如果出现警告,可能是因为部分包的版本太高的问题
测试tensorflow-gpu:
打开命令行,输入python进入python环境
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
出现类似输出结果即为安装成功
Sample Output
[name: "/cpu:0" device_type: "CPU" memory_limit: 268435456 locality { } incarnation: 4402277519343584096,
name: "/gpu:0" device_type: "GPU" memory_limit: 6772842168 locality { bus_id: 1 } incarnation: 7471795903849088328 physical_device_desc: "device: 0, name: GeForce GTX 1070, pci bus id: 0000:05:00.0" ]
3、安装CUDA、cuDNN
对于tensorflow-gpu还需另外安装CUDA、cuDNN
本人的tensorflow-gpu版本: 1.15.0
tensorflow版本:1.14.0
CUDA版本: 10.0
各位安装的时候要注意版本的匹配问题!!!
CUDA安装包百度云链接(是一个exe文件,双击安装):