一开始以为是文件路径设置的问题,接着发现是cuda的配置有误,因为刚配置了yolo-gpu,在配置过程中有改动环境变量,所以可能是要重新设置。
安装路径
/usr/local/cuda-9.0
定位 libcublas.so.9.0
:
# build search index
updatedb
# find the "missing" file
locate libcublas.so.9.0
可以看出是在
/usr/local/cuda-9.0/lib64/stubs/libcublas.so.9.0
意识到两个问题:
1.在~/.bash_profile
下设置的环境变量
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64/
这里改为
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64/stubs/
最后再赋予libcublas.so.9.0文件root权限
sudo chown -R MY_USER:MY_GROUP /usr/local/cuda-9.0/
sudo chmod -R 777 /usr/local/cuda-9.0/
OK