2016-04-05 23 views
1

https://github.com/soumith/torch-androidに示されている手順に照らして、私はtorch-androidをサーバー(Ubuntu)にビルドしたいと考えています。しかし、私はbuild.shを実行すると、スクリプトはGPU(サーバがテスラK80を持っている)を見つけることができません。torch-androidの構築方法

CMake Error at install/share/cmake/torch/FindCUDA/select_compute_arch.cmake:91 (message): 
    Only 
    Fermi;Kepler;Kepler+Tegra;Kepler+Tesla;Maxwell;Maxwell+Tegra;Pascal;All;Manual;Auto 
    architecture names are supported. 
Call Stack (most recent call first): 
    distro/extra/cutorch/lib/THC/CMakeLists.txt:62 (CUDA_SELECT_NVCC_ARCH_FLAGS) 

オープンbuild.shを、私たちは

#!/bin/bash 
# have ndk-build in your $PATH and the script figures out where your ANDROID_NDK is at. 
# optionally, modify the variables below as needed. 
NDKABI=21 
NDKVER=toolchains/arm-linux-androideabi-4.9 

ARCH=${ARCH:-"v7n"} 

if [[ "$ARCH" == "v8" ]]; then 
    APP_ABI=arm64-v8a 
    M_ARCH=-march=arm8-a 
    ABI_NAME=aarch64-linux-androideabi 
    COMPUTE_NAME=Maxwell-M 
elif [[ "$ARCH" == "v7n" ]]; then 
    APP_ABI="armeabi-v7a with NEON" 
    M_ARCH="-march=armv7-a" 
    ABI_NAME=armv7-linux-androideabi 
    COMPUTE_NAME=Kepler-M 
elif [[ "$ARCH" == "v7" ]]; then 
    APP_ABI="armeabi-v7a" 
    M_ARCH="-march=armv7-a" 
    ABI_NAME=armv7-linux-androideabi 
    COMPUTE_NAME=Kepler 
fi 

    NVCC=`which nvcc` 
export MAKE=make 
export MAKEARGS=-j$(getconf _NPROCESSORS_ONLN) 

を見つけることができ、私は私がすべきだと思いますbuild.shを修正してください。しかし、私はそれを知らないのです。誰もがこの問題を解決する方法を知っていますか?

答えて

2

これらのコマンドをternimalで実行しましたか?

  1. cd ~/torch-android;
  2. git submodule update --init --recursive;
  3. ./build;