最近在实验室的服务器上搭建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进行版本控制安装,果然,装好后,直接就能用了。
一点经验,大家有什么问题或想法可以直接在下边留言。