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

​

相关文章