链接:https://www.jianshu.com/p/bcf37d0e4e9b
为了入门机器学习的小伙伴能安装好工具,特制作此教程
-
按照 Anaconda 下载网站上的说明下载并安装 Anaconda。
-
调用以下命令创建名为 tensorflow 的 conda 环境:
conda create -n tensorflow pip python=3.6
- 发出以下命令以激活 conda 环境:
activate tensorflow
- 发出相关命令以在 conda 环境中安装 TensorFlow。要安装仅支持 CPU 的 TensorFlow 版本,请输入以下命令:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --ignore-installed --upgrade tensorflow
要安装 GPU 版本的 TensorFlow,请输入以下命令(在同一行):
pip install tensorflow-gpu==1.5.1
有个很奇怪的问题,好像cuda9.0 cudnn7.* 只能安装对应版本的tensorflow,不然就会报错
ImportError:libcublas.so.10(比如,也可能是8.0):cannot open shared object file:No such file or directory
测试tensorflow是否能在GPU上跑
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
一个更general的方法,真的被pip pip3坑了--!
很坑的是不能用pip或者pip3安装库,否则就会出现python3.5和3.6不匹配的问题而需要建立3.5的环境。应该用conda命令安装:conda install tensorflow,其他库:jupyter sklearn等等也是一样。