運行caffe時遇到cannot create Cublas handle. Cublas won't be available.錯誤

在Ubuntu16.04下,已安裝CUDA8.0和cudnn5.1,編譯caffe通過,make runtest通過。但在使用caffe進行網絡訓練時,發生錯誤如下:

解決方法:這是顯卡驅動版本過高的原因。本人的1080Ti顯卡使用NVIDIA-Linux-x86_64-384.69驅動就會出現上述問題,換成NVIDIA-Linux-x86_64-375.82版本的驅動,問題解決。附NVIDIA顯卡驅動下載地址:http://www.nvidia.cn/Download/index.aspx?lang=cn

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章