使用GPU跑
THEANO_FLAGS='floatX=float32,device=gpu' python -u lstm.py >> lstm.log&
problem
nvcc compiler not found on $path.
export PATH=/home/ubuntu/venv/bin:/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda:/usr/local/cuda/lib64:/opt/OpenBLAS-no-openmp/lib
网友评论