win10 rtx2060 安装cuda10.1 + cudnn7.6 + tensorflow2.0-gpu

1.选择合适的cuda和cudnn版本

tensorflow的github:
https://github.com/fo40225/tensorflow-windows-wheel 查看具体适配的版本,
在这里插入图片描述
我这里是要安装tf2.0,所以找到相关适配的cuda和cudnn版本:
10.1.243_426.00 / 7.6.4.38

并下载对应的tensorflow_gpu-2.0.0-cp37-cp37m-win_amd64.whl安装包,存入D盘根目录

2.下载相应版本的CUDA 和CUDNN

https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/rdp/cudnn-archive

3.安装cuda和cudnn

cuda傻瓜安装,cudnn下载之后是一个压缩包(我这里直接下载到了CUDA的安装目录,因为稍后会解压到这里):
在这里插入图片描述
把压缩包里红框的三个文件夹里的东西,
这三个

导入到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1”中对应的文件夹;

4.安装tensoflow

进入anaconda命令行,输入“pip install “D:\tensorflow_gpu-2.0.0-cp37-cp37m-win_amd64.whl””

5.测试tensoflow安装是否成功

anaconda命令行,
输入“python”回车,
import tensorflow as tf
tf.__version__
tf.__path__

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值