Cannot create Cublas handle

42 views
Skip to first unread message

Chad Cordes

unread,
Oct 20, 2018, 2:34:42 PM10/20/18
to Caffe Users
Gooday,

I am attempting to run a demo.py with my local Caffe build, but am getting the following error in python shell when attempting to set the gpu:

In python shell:
caffe.set_mode_gpu()
WARNING: Logging before InitGoogleLogging() is written to STDERR
E1020 14:06:47.083652 23219 common.cpp:114] Cannot create Cublas handle. Cublas won't be available.

Could someone kindly provide some guidance on how to address this issue?
Caffe build passes make runtest with no errors. System and build details below. Sincere thanks for any support.
---------------------------------------------------------------------------------------------------------------------------------------------------------------------

NAME="Ubuntu"
VERSION="18.04.1 LTS (Bionic Beaver)"

GPU
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 390.87                 Driver Version: 390.87                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Quadro K620         Off  | 00000000:03:00.0  On |                  N/A |
| 34%   39C    P8     1W /  30W |    234MiB /  1997MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

libcudnn.so.7 -> libcudnn.so.7.0.5
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION    (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"

Makefile.config variation from default example:
USE_CUDNN := 1
OPENCV_VERSION := 3
CUSTOM_CXX := g++-6
CUDA_ARCH := #-gencode arch=compute_20,code=sm_20 \
#               -gencode arch=compute_20,code=sm_21 \
                -gencode arch=compute_30,code=sm_30 \
                -gencode arch=compute_35,code=sm_35 \
                -gencode arch=compute_50,code=sm_50 \
                -gencode arch=compute_52,code=sm_52 \
                -gencode arch=compute_60,code=sm_60 \
                -gencode arch=compute_61,code=sm_61 \
                -gencode arch=compute_61,code=compute_61
ANACONDA_HOME := $(HOME)/anaconda3
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
                $(ANACONDA_HOME)/include/python3.6m \
                $(ANACONDA_HOME)/lib/python3.6/site-packages/numpy/core/include
PYTHON_LIBRARIES := boost_python-py36 python3.6m
PYTHON_LIB := $(ANACONDA_HOME)/lib
WITH_PYTHON_LAYER := 1
BLAS := atlas


Reply all
Reply to author
Forward
0 new messages