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

时间:2022-05-04 09:11:29

链接:https://www.jianshu.com/p/bcf37d0e4e9b

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

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

  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上跑

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等等也是一样。