服务器.bashrc配置

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

# 设置参数和训练模型
model = gensim.models.Word2Vec(sentences, size=128, window=10, min_count=5, workers=4, sg=1)