caffe powered和GPU启用Microsoft Azure VM

时间:2017-07-15 02:30:27

标签: machine-learning gpu caffe nvidia azure-dsvm

我正在尝试为Azure中的模型培训构建VM。我发现这个Data Science Virtual Machine for Linux (Ubuntu) VM似乎是一个合适的候选者。

不幸的是,当我启动VM并安装了caffe先决条件时,我无法运行测试。我在make runtest上收到了以下错误(make allmake test已完成且没有错误):

NVIDIA: no NVIDIA devices found
Cuda number of devices: 0
Setting to use device 0
Current device id: 0
Current device name: 
Note: Randomizing tests' orders with a seed of 97204 .
[==========] Running 2041 tests from 267 test cases.
[----------] Global test environment set-up.
[----------] 11 tests from AdaDeltaSolverTest/3, where TypeParam = caffe::GPUDevice<double>
[ RUN      ] AdaDeltaSolverTest/3.TestAdaDeltaLeastSquaresUpdateWithHalfMomentum
NVIDIA: no NVIDIA devices found
E0715 02:24:32.097311 59355 common.cpp:114] Cannot create Cublas handle. Cublas won't be available.
NVIDIA: no NVIDIA devices found
E0715 02:24:32.103780 59355 common.cpp:121] Cannot create Curand generator. Curand won't be available.
F0715 02:24:32.103914 59355 test_gradient_based_solver.cpp:80] Check failed: error == cudaSuccess (30 vs. 0)  unknown error
*** Check failure stack trace: ***
    @     0x7f77a463f5cd  google::LogMessage::Fail()
    @     0x7f77a4641433  google::LogMessage::SendToLog()
    @     0x7f77a463f15b  google::LogMessage::Flush()
    @     0x7f77a4641e1e  google::LogMessageFatal::~LogMessageFatal()
    @           0x7115e3  caffe::GradientBasedSolverTest<>::TestLeastSquaresUpdate()
    @           0x7122af  caffe::AdaDeltaSolverTest_TestAdaDeltaLeastSquaresUpdateWithHalfMomentum_Test<>::TestBody()
    @           0x8e6023  testing::internal::HandleExceptionsInMethodIfSupported<>()
    @           0x8df63a  testing::Test::Run()
    @           0x8df788  testing::TestInfo::Run()
    @           0x8df865  testing::TestCase::Run()
    @           0x8e0b3f  testing::internal::UnitTestImpl::RunAllTests()
    @           0x8e0e63  testing::UnitTest::Run()
    @           0x466ecd  main
    @     0x7f77a111c830  __libc_start_main
    @           0x46e589  _start
    @              (nil)  (unknown)
Makefile:532: recipe for target 'runtest' failed
make: *** [runtest] Aborted (core dumped)

是否可以使用caffe在Azure中启动适合启用GPU的机器学习的虚拟机?

有关VM here的所有详细信息 enter image description here

1 个答案:

答案 0 :(得分:3)

Ubuntu的数据科学虚拟机(DSVM)已经在/ opt / caffe中安装了Caffe。要在GPU上使用它,请通过选择其中一个NC大小来创建具有K80 GPU的VM。 (请务必选择HDD作为存储类型,否则NC尺寸将不会出现。)Caffe将随时可用。

另请注意,PyCaffe可用。在终点站:

source activate root

然后python将提供PyCaffe。