編譯安裝tensorflow1 12 2

2021-09-20 21:07:03 字數 2031 閱讀 2955

環境:

1) os

16.04.1-ubuntu

2)cpu有點年頭了

# cat /proc/cpuinfo | grep "model name"

model name    : amd athlon(tm) ii x2 250 processor

model name    : amd athlon(tm) ii x2 250 processor

3)cuda

步驟:1) cuda 解除安裝:

/usr/local/cuda/bin/*uninstall*

2) nvidia驅動解除安裝

/usr/bin/nvidia-uninstall

cuda安裝包自帶驅動,無需另外安裝

3) 安裝cuda前記得禁用nouveau

ubuntu自帶的nouveau驅動會影響cuda安裝

lsmod | grep nouveau

若有輸出,則

cd /etc/modprobe.d

sudo touch blacklist-nouveau.conf

sudo gedit blacklist-nouveau.conf

在開啟的檔案中輸入

blacklist nouveau

options nouveau modeset=0

儲存退出,然後

sudo update-initramfs -u

一定要重啟,再用終端檢測一下,按ctrl+alt+f1切換到終端,登入,init 3 切換到文字模式

lsmod | grep nouveau

4)cuda安裝, 文字模式下安裝

cuda_9.2.148_396.37_linux.run

記得不要安裝opengl,

5)設定環境變數

sudo gedit /etc/profile

末尾新增:

export path=/usr/local/cuda-9.0/bin:$path

export ld_library_path=/usr/local/cuda-9.0/lib64:$ld_library_path

nvcc --version 檢視cuda版本

6)cudnn安裝

用deb檔案安裝(失敗)

7) 現成的 tensorflow whl安裝檔案(失敗,原因是sse4.1不支援)

8) 安裝python3.6sudo add-apt-repository ppa:jonathonf/python-3.6sudo apt-get updatesudo apt-get install python3.6sudo apt-get install python3.6-dev

sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.5 1

sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 2

原文:10) bazel 從原始碼編譯

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

必須安裝 bazel 1.18.1, 最新的2.4.1是不能用的i got the same error in building tensorflow r1.9 for one older nvidia

11)得到whl檔案,安裝成功

tensorflow 1 共享變數

共享變數 reuse variables example1 with tf.variable scope try 先建立兩個變數w1,w2 w2 tf.get variable w1 shape 2,3,4 dtype tf.float32 w3 tf.get variable w2 shape 2...

tensorflow1 構建線性模型

x是給定的輸入資料 使用tensorflow構建乙個模型,開始的時候,w和b全部給成0,讓其訓練,使其接近預設的模型。即讓w接近0.1,b接近0.2 import tensorflow as tf import numpy as np x data np.random.rand 100 y data...

神經網路 tensorflow 1

import tensorflow as tf import numpy as np create data x data np.random.rand 100 astype np.float32 在tensorflow中大部分的資料的資料型別都是float32 y data x data 0.1 ...