升级tensorflow的版本,我是直接将tensorflow升级到了1.11.0,但是其需要的cuda和cudnn的版本也需要跟着升级。
参考文献:https://blog.csdn.net/weixin_32820767/article/details/80421913
一 安装cuda9.0
1. 在自己的目录下为cuda9.0新建一个文件夹,用于存放 cuda_9.0.176_384.81_linux.run
2. 生成可执行文件
chmod 777 cuda_9.0.176_384.81_linux.run
3. 执行(安装过程中的一些选项和cuda8.0一样)
./cuda_9.0.176_384.81_linux.run
4. 修改环境配置
$ sudo gedit /etc/profile
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
export LIBRARY_PATH="/usr/local/cuda/lib64:$LIBRARY_PATH"
5. 切换cuda9.0
rm –rf /usr/local/cuda
ln -s /usr/local/cuda-9.0 /usr/local/cuda
然后重新加载环境变量
$ source /etc/profile
二 安装cudnn7.0(安装方式按照cudnn6.0的安装方式)
三 安装高版本的tensorflow
pip install tensorflow==1.11.0
原文地址:https://www.cnblogs.com/lipanDL/p/9863500.html
时间: 2024-11-08 05:03:58