I also had some issues about cuda malloc and going to cuda 11 solved it if I remember correctly
this is my output :
thoth@bigC:~$ nvidia-smi
Thu Jul 30 13:17:34 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.51.06 Driver Version: 450.51.06 CUDA Version: 11.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce GTX 970 On | 00000000:01:00.0 On | N/A |
| 34% 35C P8 20W / 200W | 388MiB / 4041MiB | 1% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 959 G /usr/lib/xorg/Xorg 15MiB |
| 0 N/A N/A 1053 G /usr/bin/gnome-shell 49MiB |
| 0 N/A N/A 2062 G /usr/lib/xorg/Xorg 156MiB |
| 0 N/A N/A 2196 G /usr/bin/gnome-shell 88MiB |
| 0 N/A N/A 2803 G ...AAAAAAAAA= --shared-files 69MiB |
+-----------------------------------------------------------------------------+
thoth@bigC:~$ nvcc --version
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
thoth@bigC:~$ cat /usr/local/cuda/version.txt
CUDA Version 11.0.207