如何在Ipython笔记本上使用GPU?
操作系统:Ubuntu 14.04LTS如何在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
语言: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