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

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！