如何在Ipython笔记本上使用GPU?

如何在Ipython笔记本上使用GPU?,ipython,theano,nvcc,keras,Ipython,Theano,Nvcc,Keras,操作系统:Ubuntu 14.04LTS 语言:Python Anaconda 2.7(keras,theano) GPU:GTX980Ti CUDA:CUDA 7.5 我想使用我的GPU(GTX980Ti)在IPython笔记本上运行keras python代码 但我找不到它 我想测试下面的代码。当我在Ubuntu终端上运行它时, 我的命令如下(它使用GPU很好,没有任何问题) 首先,我设置如下路径 export PATH=/usr/local/cuda/bin:$PATH export LD

操作系统:Ubuntu 14.04LTS
语言:Python Anaconda 2.7(keras,theano)
GPU:GTX980Ti CUDA:CUDA 7.5

我想使用我的GPU(GTX980Ti)在IPython笔记本上运行keras python代码
但我找不到它

我想测试下面的代码。当我在Ubuntu终端上运行它时, 我的命令如下(它使用GPU很好,没有任何问题)

首先,我设置如下路径

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH 
THEANO_FLAGS='floatX=float32,device=gpu0,nvcc.fastmath=True'  python myscript.py
第二步,我按如下方式运行代码

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH 
THEANO_FLAGS='floatX=float32,device=gpu0,nvcc.fastmath=True'  python myscript.py

而且运行良好

但是当我在pycharm(pythonide)或 当我在Ipython笔记本上运行它时,它不使用gpu。 它只使用CPU

myscript.py代码如下所示

from theano import function, config, shared, sandbox
import theano.tensor as T
import numpy
import time
vlen = 10 * 30 * 768  # 10 x #cores x # threads per core
iters = 1000
rng = numpy.random.RandomState(22)
x = shared(numpy.asarray(rng.rand(vlen), config.floatX))
f = function([], T.exp(x))
print(f.maker.fgraph.toposort())
t0 = time.time()
for i in xrange(iters):
    r = f()
t1 = time.time()
print("Looping %d times took %f seconds" % (iters, t1 - t0))
print("Result is %s" % (r,))
if numpy.any([isinstance(x.op, T.Elemwise) for x in f.maker.fgraph.toposort()]):
    print('Used the cpu')
else:
    print('Used the gpu')
为了解决这个问题,我强制代码使用gpu,如下所示 (在myscript.py上再插入两行)

然后生成如下错误

ERROR (theano.sandbox.cuda): nvcc compiler not found on $PATH. Check your nvcc installation and try again.
怎么做???我花了两天时间

我确实使用了home目录中的“.theanoc”文件。

我在ipython笔记本上使用了theano,使用的是我系统的GPU。这种配置在我的系统上似乎运行良好。(配备GTX 750M的Macbook Pro)

My~/.TheAnoc文件:

[global]
cnmem = True
floatX = float32
device = gpu0
各种环境变量(我使用虚拟环境(macvnev):

如何运行ipython笔记本(对我来说,设备是gpu0):

$nvcc-V的输出:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Thu_Sep_24_00:26:39_CDT_2015
Cuda compilation tools, release 7.5, V7.5.19

从您的帖子中,可能是您将$PATH变量设置错误。

您是否仔细检查了
nvcc
是否位于
/usr/local/cuda/bin
?在安装过程中可能选择了不同的位置。当您从Python中打印
PATH
时,它是否包含指向
nvcc
的路径?它可能正在接收di环境变量
PATH
的实例与您添加的
nvcc
位置的实例不同。@talomies谢谢您的回答!@talomies但我再次检查/usr/local/cuda/bin…中是否有nvcc,/usr/local/cuda-7.5/bin也有nvcc。。有问题吗?!您是否将这两个
导出
comman将ds导入您的
~/.bashrc
文件?尤其是
导出路径=/usr/local/cuda/bin:$PATH
。然后重新启动?
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Thu_Sep_24_00:26:39_CDT_2015
Cuda compilation tools, release 7.5, V7.5.19