这两天有空又尝试安装gpu版本的Lasagne,不知道怎么捣鼓的就可以了。现在把它写下了,希望能给大家提供帮助。
首先,安装cpu版的吧,这个在之前的文章中:
https://blog.youkuaiyun.com/guoqiangszu/article/details/90710805
接下来,我把我安装gpu的步骤给大家说一下。
1、我安装网上的说法,安装了pygpu
conda install pygpu
pip install pygpu
都试过了,前者没安装成功,后者成功了,但配置依然有问题。
2、我以为是pygpu的版本问题,又进行了很多版本的尝试,还是不行。
3、在另一个网站看到要使用如下命令安装pygpu
conda install pygpu theano
如图所示:
装好pygpu,我又根据网上的各种说法,逐一进行尝试,添加修改了.theanorc.txt
文件,具体内容如下,
[global]
device=cuda
floatX=float32
force_device=True
CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
[nvcc]
fastmath = True
[dnn]
enabled = True
include_path=C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0/include
library_path=C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v9.0/lib/x64
其实,我也并不清楚到底是哪个地方让我成功安装了gpu环境,只能是一个一个试,最后成功就成功了。很不容易啊!!!