1 2 3 4

tensorflow-gpu安装

由于最近在学习tensorflow,需要在windows环境下安装tensorflow,刚开始装出现了各种报错,基本都是版本不对应,后来重新安装一遍,可以使用了,这里记录一下。

我没有使用anaconda安装python,直接使用pip进行安装的。

  • 首先win+R,运行cmd,运行如下命令
pip install tensorflow-gpu==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

我这里使用的是python 3.7版本,安装的是tensorflow-gpu的2.3.0版本。

  • 接下来需要安装CUDA和CuDNN,这里需要选择对应的安装版本,版本对应的关系如下图(图片来自于水印)

然后下载完成后,一路Next安装就可以了。

  • 安装完成后,需要将CUDA对应的路径配置到环境变量,我的默认安装路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1,分别划分红框中的三个以及根路径添加进环境变量;

  • 接下来需要下载CuDnn,到官网https://developer.nvidia.com/cudnn,选择对应的7.6版本下载,下载时需要进行一个简单的注册和调研,下载后进行解压,

  • 解压后,找到cuda安装目录中对应的三个文件夹,分别将cudnn中的三个文件夹内容复制(注意这里不是直接复制文件夹,要进入文件夹,复制dll文件),然后粘贴到对应的cuda目录中的三个文件夹中。

安装完成后就基本完成了,打开cmd,运行nvcc -V查看cuda安装版本:

nvcc -V

运行nvidia-smi,查看显卡信息和gpu进程:

在python 命令行输入:

import tensorflow as tf
print(tf.test.is_gpu_available())

则显示 gpu可用,说明安装成功。

posted @ 2022-03-24 11:23  Uniqe  阅读(169)  评论(0编辑  收藏  举报