RTX2080Ti+Win10+Cuda10.1+Cudnn7.6+tensorflow-gpu2

1.tf官网说明

https://www.tensorflow.org/install/gpu
在这里插入图片描述

2.Anaconda安装

https://www.anaconda.com/products/individual

3.conda tf虚拟环境

conda create -n tf2 python=3.6.5

conda 换国内的镜像源

4.tf-gpu安装

conda install tensorflow-gpu  
#会自动安装cuda10.1,cudnn7.6.5等依赖哦,缺点是只能在此虚拟环境下使用
#或者pip,lunix系统实测有效
pip install tensorflow-gpu==2.3.0
  1. 如果报错compat,把下面版本降低一点
conda install tensorflow-estimator==2.1.0 
  1. 测试
import tensorflow as tf
tf.__version__

至此tf-gpu安装完毕,是不是觉得太简单了?
缺点有下:
该方法只适合特定环境和特定tf版本gpu,有很多版本或者用pip安装时是只安装tensorflow-gpu而不会安装cuda,cudann等依赖的

。。。

下面尝试一般方法:

5.cuda安装

  1. 为了便于其他地方使用gpu,也把cuda给安装了

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

  1. 为了装cuda,也需要VS(若有无需)
    VS官网2019

6.cudann安装

cuda安装教程+cudnn安装教程—Hello_World_admin
此文很详细了,感谢开源!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值