Сборка Torch7 с помощью Cuda 10.0 - PullRequest
Новая
       5

Сборка Torch7 с помощью Cuda 10.0

0 голосов
/ 10 февраля 2020

Моя linux машина имеет 2 CUDA, 7,5 и 10,0. Когда я пытался собрать Torch7, машина говорит мне 

Building on 40 cores
-- Found Torch7 in /home/onurberk/torch/install
-- Removing -DNDEBUG from compile flags
-- TH_LIBRARIES: TH
-- Found gcc >=5 and CUDA <= 7.5, adding workaround C++ flags
-- MAGMA not found. Compiling without MAGMA support
-- Autodetected CUDA architecture(s): 6.1 6.1 6.1 6.1
-- got cuda version 7.5
-- Found CUDA with FP16 support, compiling with torch.CudaHalfTensor
-- CUDA_NVCC_FLAGS: -gencode;arch=compute_61,code=sm_61;-DCUDA_HAS_FP16=1
-- THC_SO_VERSION: 0
-- Configuring done
-- Generating done

Так что он, безусловно, находит неправильную CUDA и выдает мне ошибку: 

nvcc fatal   : Unsupported gpu architecture 'compute_61'

Я думаю, чтобы исправить это, я нужно скомпилировать с cuda 10.0. Поэтому я пробовал разные вещи, такие как запуск в папке основного факела 

sudo cmake -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-10.0/ .

или установка переменной PATH 

 export PATH=/usr/local/cuda-10.0/bin:${PATH}

или 

 export PATH=:${PATH}:/usr/local/cuda-10.0/bin

Ничего не работает Любая идея или помощь? Мои cuda и cuda-10.0 находятся в папке / usr / local.

...