Tensorflow(GPU版本)环境部署

由tensflow官网查询tensorflow、python、cuda、cuDNN版本对应关系

https://www.tensorflow.org/install/source_windows

Anconda包安装

根据所需程序确定tensorflow版本

创建环境,并安装python包

conda create -n tensorflow_cpu python=3.6

conda create -n tensorflow_gpu_1 python=3.6

tensorflow安装(镜像安装)

Anconda包管理

  • conda list 列举当前环境下的所有包
  • conda search packagename 搜索某包
  • conda updata packagename 更新当前环境某包
  • conda/pip remove/uninstall packagename 删除当前环境某包(安装的时候是用pip 安装, 卸载的时候也需要pip uninstall)
  • conda env export > environment.yaml 环境会被保存在 environment.yaml文件中(到c盘用户里找)。

安装CUDA

直接谷歌,例如cuda9,因为有的就版本在官网里也挺难找的。安装步骤直接按部就班安装就是。

建议默认路径(或者记住路径) C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1,方便后面添加环境变量。

  • 检查是否安装成功

    • 系统会多两个系统变量

    image-20220711221558454

    • 重启电脑后,进入cmd界面,输入nvcc -V显示

      在这里插入图片描述

安装cuDNN

同样直接谷歌,例如cuDNN,找到对应CUDA版本下载。

下载的cuDNN压缩包解压后出现如下三个文件夹子

在这里插入图片描述

将里面文件复制到CUDA安装路径对应文件夹中。

image-20220711221901034

添加环境变量

以下四个环境变量尽量都添加

image-20220711222517114

检查是否安装成功

https://blog.csdn.net/jhsignal/article/details/111398427

版本匹配示例(已实验)

  • tensorflow_gpu1.5.0
  • python3.6
  • CUDA9.0
  • cuDNN for CUDA9.0 (v7.6.5.32)
posted @ 2022-07-11 23:06  xfw121  阅读(131)  评论(0)    收藏  举报