记一次装GPU环境

  1. 首先是装显卡驱动,显卡是英伟达的tesla v100,从官网上下载了驱动,系统是centos 7.5, 安装的时候首先提示安装kernel-devel,yum install即可,然后说找不到系统对应的源码,原来是下载的源码和系统版本对应不上,yun update kernel即可,紧接着又报gcc版本问题,说包的kernel是4.8.5 但是当前的gcc版本是6.3.1,出去gcc --version一看,是4.8.5啊,一脸懵逼,以为是自己理解错了,把gcc搞成了6.3.1的,结果一样的报错,一想是环境变量CC,cmake文件中的c编译器,一看果然是指向了6.3.1的gcc,改了之后编译到中间又出错,说不能加载nvidia.ko文件,导致这个的原因可能是已经有其他驱动了或者gcc版本不对,或者是和显卡不匹配,问题排查后只有gcc这个可能有问题,自然想到是编译c++代码的g++,看环境变量CXX发现果然是6.3.1的,改正后顺利完成,
  2. 安装cuda和cudnn cuda我下载的rpm包,yum localinstall + yum install搞定,cudnn解压拷贝即可
  3. 安装TensorFlow, 一开始直接pip install tensorflow==[version]了,结果是cpu版本的,pip install tensorflow-gpu
  4. 还是不行,找不到so文件,把cuda的lib加到LD_LIBRARY_PATH即可
  5. gcc下载地址http://ftp.gnu.org/
  6. pip install xxx -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值