CUDA driver version is insufficient for CUDA runtime version

After I ran ./demo_test.sh, returns:
`[2020-04-22 20:12:02,131 INFO test.py line 172 28455] Loaded configuration file config/ade20k-resnet50dilated-ppm_deepsup.yaml
[2020-04-22 20:12:02,131 INFO test.py line 173 28455] Running with config:
DATASET:
imgMaxSize: 1000
imgSizes: (300, 375, 450, 525, 600)
list_train: ./data/training.odgt
list_val: ./data/validation.odgt
num_class: 150
padding_constant: 8
random_flip: True
root_dataset: ./data/
segm_downsampling_rate: 8
DIR: ade20k-resnet50dilated-ppm_deepsup
MODEL:
arch_decoder: ppm_deepsup
arch_encoder: resnet50dilated
fc_dim: 2048
weights_decoder:
weights_encoder:
TEST:
batch_size: 1
checkpoint: epoch_20.pth
result: ./
TRAIN:
batch_size_per_gpu: 2
beta1: 0.9
deep_sup_scale: 0.4
disp_iter: 20
epoch_iters: 5000
fix_bn: False
lr_decoder: 0.02
lr_encoder: 0.02
lr_pow: 0.9
num_epoch: 20
optim: SGD
seed: 304
start_epoch: 0
weight_decay: 0.0001
workers: 16
VAL:
batch_size: 1
checkpoint: epoch_20.pth
visualize: False
THCudaCheck FAIL file=torch/csrc/cuda/Module.cpp line=32 error=35 : CUDA driver version is insufficient for CUDA runtime version
Traceback (most recent call last):
File "test.py", line 198, in
main(cfg, args.gpu)
File "test.py", line 95, in main
torch.cuda.set_device(gpu)
File "/home/kai/anaconda3/envs/hrnet/lib/python3.6/site-packages/torch/cuda/__init__.py", line 262, in set_device
torch._C._cuda_setDevice(device)
RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at torch/csrc/cuda/Module.cpp:32
`
Pytorch=0.4.1, cuda80, python=3.6.
Can anyone help me out?