安装theano

安装的是theano 0.8.2版本直接anaconda安装的,在遇到GPU运行的时候遇到了点问题

vim ~/.theanorc 之后设置

[global]
device=gpu
floatX=float32  ## 这里一定要是float32 我就是一直设置的float64所以出错,用不了GPU
root=/usr/local/cuda-8.0
[nvcc]
fastmath = True
[blas]
ldflags = -lopenblas
[cuda]
root = /usr/local/cuda-8.0
[lib]
cnmem = 1

还需要设置.bashrc文件 

export PATH="/usr/local/cuda-8.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH"
export LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$LIBRARY_PATH"

 

posted @ 2018-04-02 14:47  Emma00  阅读(130)  评论(0编辑  收藏  举报