windows10使用conda安装 gpu版本的tensorflow2.0

提示:已经提前安装Anaconda

一、本机配置

  • 系统版本:windows 10 64位
  • 显卡:GeForce RTX 2060

对于NVIDIVA的显卡,其CUDA 算力要求不低于3.0
NVIDIA驱动程序需要410.x或更高版本
CUDA版本:10.0
cudnn版本号需要不小于7.4.1

算力参考
NVIDIA驱动程序版本查看:nvidia-smi(命令行执行)
NVIDIA驱动程序下载地址

二、安装CUDA和cudnn

因为我的NVIDIA驱动程序是符合要求的,所以就没有更新,需要更新的同学可以点击上面的链接下载

1. 打开Anaconda Prompt

2. 安装CUDA

命令行输入

conda install cudatoolkit=10.0.130

推荐安装版本:10.0.130

3. 安装cudnn

命令行输入

conda install cudnn=7.6.0

推荐版本:7.6.0

4. 安装tensorflow

命令行输入

conda install tensorflow-gpu=2.0.0

5. 检测是否安装成功

命令行打开python,输入

import tensorflow as tf
tf.test.is_gpu_available()

猜你喜欢

转载自blog.csdn.net/qq_45465526/article/details/108299866