1. 程式人生 > >【tensorflow】tensorflow 常見問題記錄

【tensorflow】tensorflow 常見問題記錄


問題1: libcublas.so.9.0: cannot open shared object file: No such file or directory

解決方案:tensorflow 版本與cuda版本不相容問題,正確版本對映關係如下:

tensorflow-gpu版本 cuda版本
tensorflow 1.6/1.5 cuda9.0
tensorflow 1.3/1.4 cuda8.0

處理方法:最好的方式是更改tensorflow的版本

#pip工具,適配cuda9.0
pip install tensorflow-gpu==1.6.0  
#如果需要刪除歷史版本可用如下命令
sudo pip install -U --pre tensorflow-gpu==1.6.0

#conda工具,適配cuda9.0
conda install tensorflow-gpu==1.6.0