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

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

转载于:https://www.cnblogs.com/BambooEatPanda/p/10510445.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值