在超算上投tensorflow GPU任務

2021-08-22 11:33:44 字數 1575 閱讀 1455

export ld_library_path=/public/software

/compiler/cuda

/7/cuda-8.0/lib64/:

$ld_library_path

export include=/public/software

/compiler/cuda

/7/cuda-8.0/include:

$include

export path=/public/software

/compiler/cuda

/7/cuda-8.0/bin:

$path

export ld_library_path=/public/home

/yangbo1/cudnn

/cuda/lib64:

$ld_library_path

export include=/public/home

/yangbo1/cudnn

/cuda/include

:$include

#pbs -l nodes=1:ppn=1:gpus=1 #《節點數目,cpu數目,gpu數目》

#pbs -q 《佇列名稱,確保這個佇列有gpu>

#pbs -l walltime=99:23:59:59

#pbs -n 《任務名稱》

cd $pbs_o_workdir

nvidia-smi > nvdebug_s #《檢視gpu裝置》

# 把並行執行的模組加上

source //public/software/compiler/intel/composer_xe_2015.2.164/bin/compilervars.sh intel64

source //public/software/mpi/intelmpi/5.0.2.044/intel64/bin/mpivars.sh

# 載入cuda8.0

export ld_library_path=/public/software/compiler/cuda/7/cuda-8.0/lib64/:$ld_library_path

export path=/public/software/compiler/cuda/7/cuda-8.0/bin:$path

export include=/public/software/compiler/cuda/7/cuda-8.0/include:$include

# 載入cudnn

export ld_library_path=/public/home/yangbo1/cudnn/cuda/lib64:$ld_library_path

export include=/public/home/yangbo1/cudnn/cuda/include:$include

# 檢查環境變數

env > env_check

export cuda_visible_devices=0

# 執行main.py開始任務,輸出到print_out中

python main.py > print_out

Tensorflow 七 超引數搜尋

一 超引數 超引數就是每次訓練中提前設定好的值,在訓練過程不會隨著data的輸入而變化,並且會影響結果的引數。比如 網路結構引數 層數,每層寬度,啟用函式等 訓練引數 batch size,學習率等 在深度學習中,超引數眾多,逐一除錯會很繁瑣,所以需要輔助可以迅速定位超引數。二 超引數搜尋策略 超引...

tensorflow在windows環境下的安裝

2.開啟電腦的vt 虛擬化技術 3.然後安裝docker 參考以下兩個教程 安裝完docker之後,發現tensorflow裝不上,所以現在又要先把docker學習一下。所以我離目標越來越遠了。我的歷程 1 要做建築物識別,先做深度學習,matlab不能滿足需求,選擇現下最火的tensorflow ...

在Mac OS X上安裝TensorFlow

sudo easy install pip如果這部出現錯誤,可以看看自己 usr bin 目錄是否有多個版本的 easy install,比如說我就有三個,可以試試換個 easy install版本試試,當時用easy install時遇到錯誤了,後面換成easy install 2.7就好了 vi...