自我教训:去年帮自己的笔记本配置这些环境的时候没有把这些步骤记录下来,搞得又花了一天的时间来配置环境。奇怪的是不知道为什么突然我的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