由 caffe 驱动并启用 GPU 的 Microsoft Azure VM

Question

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

不幸的是，当我启动 VM 并安装 caffe 先决条件时，我无法运行测试。我在 make runtest 上收到以下错误（make all 和 make 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 机器学习的虚拟机？

关于虚拟机的所有详细信息here

Answer 1

Ubuntu 的 Data Science Virtual Machine (DSVM) 已经在 /opt/caffe 中安装了 Caffe。要在 GPU 上使用它，请通过选择其中一种 NC 大小来创建具有 K80 GPU 的 VM。（一定要选择HDD作为存储类型，否则NC尺寸不会出现。）Caffe将开箱即用。

另请注意，PyCaffe 可用。在终端：

source activate root

然后 python 将有 PyCaffe 可用。

由 caffe 驱动并启用 GPU 的 Microsoft Azure VM

caffe powered and GPU enabled Microsoft Azure VM

machine-learning

gpu

caffe

nvidia

azure-dsvm