Я запускаю проект DL4J на своем ноутбуке, используя мой мобильный графический процессор.Он работает нормально, GPU обнаружен, и мои модели проходят обучение.Но после переключения на рабочий стол с дискретной графикой, эта же программа пытается использовать процессор!Обе машины работают под управлением Ubuntu 18.04 и имеют установленную cuda 9.0.
При запуске программы я получаю вывод с ноутбука:
ИНФОРМАЦИЯ: загружен [JCublasBackend] бэкэнд, дек 05,2018 3:26:33 AM org.nd4j.nativeblas.NativeOpsHolder INFO: Количество потоков, использованных для NativeOps: 32 декабря 05, 2018 3:26:33 org.nd4j.nativeblas.Nd4jBlas INFO: Количество потоков, используемых для BLAS:0 дек. 05, 2018 3:26:33 утра org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner printEnvironmentInformation ИНФОРМАЦИЯ: Используемая внутренняя часть: [CUDA];ОС: [Linux] 05 декабря 2018 3:26:33 утра org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner printEnvironmentInformation INFO: Ядра: [8];Память: [3,4 ГБ];05 декабря 2018 3:26:33 утра org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner printEnvironmentInformation ИНФОРМАЦИЯ: Поставщик Blas: [CUBLAS] 05 декабря 2018 3:26:33 org.nd4j.linalg.jcublas.ops.executioner.CudaExecutioner printEnvironmentInformation ИНФОРМАЦИЯ: Имя устройства: [GeForce GTX 960M];СС: [5,0];Общая / свободная память: [4242604032]
И это с рабочего стола:
03: 29: 44.843 [main] INFO org.nd4j.linalg.factory.Nd4jBackend- Загруженный [CpuBackend] backend
Сбой происходит сразу после того, как мои зависимости включают в себя nd4j-cuda-9.0-platform и не nd4j-native-platform.
Я почти уверенcuda работает на рабочем столе, я без проблем запускаю на нем программы tenorflow-gpu.
Я использую gradle, вот раздел зависимостей моего build.gradle:
compile group: 'org.deeplearning4j', name: 'deeplearning4j-core', version: '1.0.0-beta'
compile group: 'org.nd4j', name: 'nd4j-cuda-9.0-platform', version: '1.0.0-beta'
compile group: 'org.datavec', name: 'datavec-api', version: '1.0.0-beta'
compile group: 'org.deeplearning4j', name: 'rl4j-core', version: '1.0.0-beta'
compile group: 'org.deeplearning4j', name: 'rl4j-api', version: '1.0.0-beta'
compile group: 'org.deeplearning4j', name: 'deeplearning4j-cuda-9.0', version: '1.0.0-beta'
compile group: 'org.deeplearning4j', name: 'deeplearning4j-ui_2.10', version: '1.0.0-beta'
Я мог бы пропустить что-то простое здесь.Любая помощь с благодарностью.