CUDA9.0+tensorflow-gpu1.8.0+Python2.7服务器环境搭建经验

最近在实验室的服务器上搭建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进行版本控制安装,果然,装好后,直接就能用了。

 

一点经验,大家有什么问题或想法可以直接在下边留言。

原文地址:https://www.cnblogs.com/yourcool/p/10521955.html