在linux上cuda9.0 cudnn7.* 安装python3.6 tensorflow 1.5.1

连接:https://www.jianshu.com/p/bcf37d0e4e9bpython

 

为了入门机器学习的小伙伴能安装好工具,特制做此教程markdown

  1. 按照 Anaconda 下载网站上的说明下载并安装 Anaconda。app

  2. 调用如下命令建立名为 tensorflow 的 conda 环境:机器学习

conda create -n tensorflow pip python=3.6
  1. 发出如下命令以激活 conda 环境:
activate tensorflow
  1. 发出相关命令以在 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上跑post

 

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等等也是同样。网站

相关文章
相关标签/搜索