使用Anaconda搭建TensorFlow-GPU环境

前言:

对于深度学习来说,各种框架torch,caffe,keras,mxnet,tensorflow,pandapanda环境要求各一,如果我们在一台服务器上部署了较多的这样的框架,那么各种莫名的冲突

会一直伴随着你,吃过很多次亏之后,慢慢的接触了Anaconda,真的是很爽的一个功能,来管理环境配置。我们进行tensorflow安装的时候,还是使用Anaconda,鉴于国内墙太高

这里记录下linux的安装方式:

大概会出现这些信息:

我们选择其中的一个进行安装之前,先查询这个分支的URL路径:

5.2 安装

conda install --channel https://conda.anaconda.org/nehaljwani tensorflow-gpu

5.3 检测是否安装成功:

如果没有报错,则说明幸运的安装成功了~

对于失败的情况,我这里给出最容易出现的:

这种问题,是说我们没有找到这个动态库,或者干脆就没有这个动态库.

解决方法:

先问是不是: 输入这条命令查查看有没有: locate libcusolver.so

我们发现我们只有libcusolver.so.8.0,并没有我们要找的libcusolver.so.7.5,看了一下官方的文档:

给出的建议是: 可以使用.8.0来替代.7.5,我们命名一个.8.0的软连接为.7.5

我们先到/usr/lib/cuda/lib64 下:

然后在.bashrc系统环境下配置一下这个路径:

然后在测试:

正常了,说明已经完全安装好了~

参考:

https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

http://www.jianshu.com/p/7be2498785b1

https://stackoverflow.com/questions/42013316/after-building-tensorflow-from-source-seeing-libcudart-so-and-libcudnn-errors

https://github.com/tensorflow/tensorflow/issues/1501

Original: https://www.cnblogs.com/gongxijun/p/7194705.html
Author: 龚细军
Title: 使用Anaconda搭建TensorFlow-GPU环境

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/9517/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

免费咨询
免费咨询
扫码关注
扫码关注
联系站长

站长Johngo!

大数据和算法重度研究者!

持续产出大数据、算法、LeetCode干货,以及业界好资源!

2022012703491714

微信来撩,免费咨询:xiaozhu_tec

分享本页
返回顶部
最近整理资源【免费获取】:   👉 程序员最新必读书单  | 👏 互联网各方向面试题下载 | ✌️计算机核心资源汇总