最近在實驗室的服務器上搭建Tensorflow,CUDA是默認的9.0,管理員已經裝好了,同時環境變量都已經配好。
直接用Anaconda創建的虛擬環境,使用pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple(推薦使用清華的源)裝好了tensorflow后,發現當import tensorflow后,報錯:ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory。
但是奇怪了,在另一個Python3.5的虛擬環境中,tensorflow這樣裝就能直接用啊。
於是想到,可能是版本的問題,因此,改用pip install tensorflow-gpu==1.8 -i https://pypi.tuna.tsinghua.edu.cn/simple對tensorflow進行版本控制安裝,果然,裝好后,直接就能用了。
一點經驗,大家有什么問題或想法可以直接在下邊留言。