Theano缺少nvcc编译器

时间:2016-05-27 17:19:53

标签: environment-variables ubuntu-14.04 gpu theano

我使用以下命令在ubuntu 14.04上安装theano:

sudo apt-get install python3-numpy python3-scipy python3-dev python3-pip python3-nose g++ libopenblas-dev git

sudo pip install Theano

我已经安装了cuda7.5。如果我运行ncvv --version,我会收到以下消息:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Tue_Aug_11_14:27:32_CDT_2015
Cuda compilation tools, release 7.5, V7.5.17

我不能使用theano gpu版本。我的命令和错误消息在这里:

$sudo THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python test1.py

ERROR (theano.sandbox.cuda): nvcc compiler not found on $PATH. Check your nvcc installation and try again.

[Elemwise{exp,no_inplace}(<TensorType(float32, vector)>)]
Looping 1000 times took 2.507168 seconds
Result is [ 1.23178029  1.61879337  1.52278066 ...,  2.20771813  2.29967761
  1.62323284]
Used the cpu

2 个答案:

答案 0 :(得分:1)

首先,检查并查看在&#39; / usr / local / cuda / bin /&#39;中是否存在名为NVCC的文件。如果继续使用以下命令,如果没有,我想你必须安装CUDA工具包等。

仅运行以下命令:

$ export PATH="/usr/local/cuda/bin/:$PATH"
$ source .bashrc

您可能还必须重新启动,但即使您不重新启动它也应该可以正常工作。

答案 1 :(得分:-1)

您需要将cuda编译器的文件夹添加到PATH变量:

export PATH=/usr/local/cuda-7.5/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH

然后运行ldconfig