ubuntu16.04的caffe测试mnist出现cannot create cublas handle.cublas won't be available

刚装完的caffe编译通过后,用mnist测试出现了如下问题
ubuntu16.04的caffe测试mnist出现cannot create cublas handle.cublas won't be available

网上查找后,有的说是驱动问题,在没有更改驱动的情况下,将命令
./examples/mnist/train_lenet.sh换成
sudo ./examples/mnist/train_lenet.sh
如果仍然出现问题,就只有更换驱动版本了