首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >AWS BitFusion实例ImportError(“未找到Cuda。无法取消挑选CudaNdarray")

AWS BitFusion实例ImportError(“未找到Cuda。无法取消挑选CudaNdarray")
EN

Stack Overflow用户
提问于 2016-03-10 01:37:54
回答 1查看 650关注 0票数 0

我正在尝试解选一个CudaNdarray,但我收到了一个导入错误。我知道我需要在一台有图形处理器的机器上,所以我到处搜索,找到了this BitFusion ec2 instance,它应该有图形处理器支持和cuda。我下载了我的代码的所有依赖项,但我得到的是:

代码语言:javascript
复制
File "predictor.py", line 112, in <module>
    predict = loadModel(filePath = '2_21_dim20000PREDICT113000.save')
  File "predictor.py", line 91, in loadModel
    model = pickle.load(file2open, encoding='latin1')
  File "/home/ubuntu/anaconda3/lib/python3.5/site-packages/theano/sandbox/cuda/type.py", line 591, in CudaNdarray_unpickler
    raise ImportError("Cuda not found. Cannot unpickle CudaNdarray")
ImportError: Cuda not found. Cannot unpickle CudaNdarray

下面是我的代码(Python3):

代码语言:javascript
复制
import os
import sys
import pickle
from random import shuffle
from copy import copy

def loadModel(filePath):
    file2open = open(filePath, 'rb')
    model = pickle.load(file2open, encoding='latin1')
    b.encode("latin1")
    file2open.close()
    return zlib.decompress(b)

predict = loadModel(filePath = '2_21_dim20000PREDICT113000.save')
EN

回答 1

Stack Overflow用户

发布于 2016-04-02 03:13:05

我有几个你可以尝试的东西:

A)确保安装了Cuda工具包:

代码语言:javascript
复制
conda list | grep cudatoolkit

如果您没有看到,请尝试执行以下操作:

代码语言:javascript
复制
conda install cudatoolkit

python
>>> from numbapro import cuda
>>> cuda.is_available()
True
>>> cuda.__file__
'/home/ubuntu/anaconda3/envs/snowflakes/lib/python3.5/site-packages/numba/cuda/__init__.py'

B)是否需要在Theano下导入CUDA包?

代码语言:javascript
复制
python
>>> from theano.sandbox import cuda
>>>cuda.cuda_path
'/home/ubuntu/anaconda3/envs/snowflakes/lib/python3.5/site-packages/theano/sandbox/cuda'
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/35898716

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档