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,这里需要选择对应的安装版本,版本对应的关系如下图(图片来自于水印)
- 先安装CUDA,到其官网https://developer.nvidia.com/cuda-downloads,选择对应的10.1版本安装,
然后下载完成后,一路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可用,说明安装成功。