Ubuntu16.04、GTX850M、cuda9.0、tensorflow1.5、cuddn7.0.5环境配置

自我教训:去年帮自己的笔记本配置这些环境的时候没有把这些步骤记录下来,搞得又花了一天的时间来配置环境。奇怪的是不知道为什么突然我的GTX850M在原来的配置下不工作了(总显示显存溢出,可用的显存就30多M),把网上所有关于解决这问题的方法都试过了一遍,但没有解决问题,只有把cuda8.0升级为9.0,Tensorflow升级为1.5了,最后能够正常的工作。

nvidia驱动的安装:因为之前配置过一次环境了,所以我里面已经有NVIDIA的驱动了,驱动的版本为384.1.1,具体的驱动安装这里就不赘述了,详情可以自己搜索一下其他博客。

CUDA9.0安装:之前用到的是.CUDA8.0版本的,但是tensorflow1.5版本是需要.CUDA9.0的,这边我采用的是先卸载了原来的CUDA8.0。卸载方式如下:

我是直接找到了/usr/local/cuda/bin下的 uninstall_cuda_8.0.pl,然后在终端运行这个文件

cd /usr/local/cuda/bin
sudo ./uninstall_cuda_7.5.pl

卸载完了之后就是安装CUDA9.0,去英伟达的官网在右上角搜索“CUDA Toolkit 9.0 Download”,然后按照自己的系统选择需要下载的run文件,我下载的是:cuda_9.0.176_384.81_linux.run。

运行该文件进行安装:

sudo sh cuda_9.0.176_384.81_linux.run

在安装的时候会问你要不要安装英伟达的驱动千万记住要选择N!千万记住要选择N!千万记住要选择N!

安装完事之后声明环境变量:

# the path for cuda
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值