tensorflow環境部署

2021-08-19 10:50:21 字數 1068 閱讀 7400

準備工作:

含gpu的伺服器1臺

ubuntu14.04或ubuntu16.06 x64

gpu對應的最新版本驅動

cuda-9.0或cuda-9.1

系統自帶的python2或python3

與cuda和os版本對應的cudnn

裝置連線網際網路

pycuda

環境配置:

4028gr-tr2

e5-2680v4 * 2

記憶體256gb

注意事項:tensorflow有對應的cudnn版本,不要安裝成錯誤的版本,不然會報相容錯誤。

一、安裝步驟

1、安裝好作業系統、gpu驅動、cuda

2、設定環境變數

然後再python中import tensorflow發現會報錯:

importerror: libcublas.so.9.0: cannot open shared object file: no such file or directory

這個需要安裝正確的cudnn版本(例如cudnn7.0.5)

將cuda路徑加到path變數中:

export cuda_path="/home/user/cuda-9.1"

export cuda_bin_path=$cuda_path

export cuda_toolkit_root_dir=$cuda_path

export path=$cuda_path/bin:$path

export ld_library_path=$cuda_path/lib64:$ld_library_path

export libraray_path=$cuda_path/lib64:$library_path

export cpath=$cuda_path/include:$cpath

環境部署 Redis環境部署

目錄 redis環境部署 準備部署目錄 修改配置檔案 啟動服務 檢查 附錄redis常用指令 解壓後目錄基本如下 編譯後在src目錄中可看到編譯產物 可執行的二進位制檔案 主要是這幾個 redis server redis sentinel redis cli redis benchmark red...

Tensorflow模型部署教程

1 吳恩達團隊tensorflow2.0資料和部署第一課 基於瀏覽器的模型與tensorflow.js 2 吳恩達團隊tensorflow2.0資料和部署第二課 基於裝置的模型與tensorflow lite 3 吳恩達團隊tensorflow2.0資料和部署第三課 基於tensorflow的資料管...

TensorFlow 環境配置

1.tensorflow官方環境配置 2.tensorflow簡要教程系列 一 mac安裝tensorflow 按照官方的方法走,我發現,第一步我就走不下去。我是macos系統,第一步是安裝pip。命令是 sudo easy install pip 結果是卡在那,乙個晚上都沒結果,下面是截圖 然後各...