提示:已经提前安装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()