cuda路徑
# cuda
export LD_LIBRARY_PATH=/export/tools/cuda-9.0/lib64:$LD_LIBRARY_PATH
# cudnn
export LD_LIBRARY_PATH=/export/tools/cudnn-7.0/lib64:$LD_LIBRARY_PATH
# mkl for lpc synthesizer
export LD_LIBRARY_PATH=/export/tools/mkl/lib/intel64/:$LD_LIBRARY_PATH
export CUDA_HOME=/export/tools/cuda-9.0
export PATH="/export/tools/cuda-9.0/bin:$PATH"
locale
export LANGUAGE=en_US.UTF-8
export LC_ALL=en_US.UTF-8
tmux
if [ -z "$TMUX" ]; then
tmux attach -t 0default || tmux new -s 0default
fi
tensorflow-gpu安裝:
pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.9.0-cp35-cp35m-linux_x86_64.whl
# 設(shè)置參數(shù)和訓(xùn)練模型
model = gensim.models.Word2Vec(sentences, size=128, window=10, min_count=5, workers=4, sg=1)
