安装cuda9.0 + cudnn7.0 与 cuda8.0共存

升级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

猜你喜欢

转载自www.cnblogs.com/lipanDL/p/9863500.html