ubuntu16.04安装tf请跳转:https://blog.csdn.net/weixin_44341962/article/details/91350991
- 说明
- cuda8.0配置
- cudnn5.1配置
- 使用anaconda搭建python3.5环境
- tensorflow1.2配置
- 在pycharm中导入配置好的环境
1.说明
(1)在配置cuda前检查你的显卡驱动是否安装好
nvidia-smi
显示如下,说明显卡驱动是安装好的
(2)配置环境一定要配置相对应的版本,不然会有未知错误,本文cuda8.0+cudnn5.1+tensorflow1.2-gpu为例
2.cuda8.0配置
(1)官网下载CUDA8.0 https://developer.nvidia.com/cuda-downloads
选择Legacy Releases进入历史版本
这里我们选择CUDA Toolkit 8.0 GA2下载
(2)安装
下载完成后,打开安装程序,安装路径默认或者自定义都可以
等待CUDA Setup到100%,检查系统的兼容性,同意许可协议
选项选择自定义,因为我们有显卡驱动,精简安装会安装驱动导致不必要的麻烦
关于自定义下安装的选项,建议将CUDA和Other components下的选项全部勾选,其余不选,下图并非我安装时的选择。