【tensorflow】在GPU服務器上配置tensorflow-gpu遇到的一些坑

環境:ubuntu16.04GPU服務器,cuda9.0 、cudnn7.0.5,Anaconda3虛擬環境下python3.6

1、最開始用最常規安裝方法,pip install tensorflow-gpu=1.9.0和conda install tensorflow-gpu=1.9.0遇到網絡問題,裏面的一些包根本下載不下來。

 

2、另闢蹊徑,選擇用清華鏡像:https://mirrors.tuna.tsinghua.edu.cn/,但進來卻是這個樣子

3、請進這個網址:https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/  看到柳暗花明一片片。。。。。

4、然後隨便選擇了一個版本安裝,選擇了tensorflow-gpu1.3.0,安裝成功,真是喜極而泣!!

pip install https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp36-cp36m-linux_x86_64.whl

5、然卻慘遭雷劈、、、、、

開始以爲環境變量中沒有相關配置,因此:


$ vim ~/.bashrc 
$ export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64”  #添加該行
$ export CUDA_HOME=/usr/local/cuda #添加該行
$ source ~/.bashrc #添加結束後執行,使得配置生效

但是仍然不行,絕望至極!!!

6、查找原因,參考博客:https://blog.csdn.net/yeler082/article/details/80943040  

7、回想起系統裝的cuda是9.0,,,好吧!!!卸了重裝!!

pip uninstall tensorflow-gpu==1.3.0

pip install https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.5.0-cp36-cp36m-linux_x86_64.whl

成功!!

####################下面安裝cv2,輸入pip install opencv-python 卻瘋狂報這個錯####################

無奈,只能再次另闢蹊徑!!!關於這個問題,會在另一篇“【python-opencv】Anaconda3虛擬環境中配置cv2遇到的問題”博文中介紹!!!

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