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