在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等等也是同樣。網站

相關文章
相關標籤/搜索