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

tensorflow的github:
https://github.com/fo40225/tensorflow-windows-wheel 查看具体适配的版本,
win10 rtx2060 安装cuda10.1 + cudnn7.6 + tensorflow2.0-gpu
我这里是要安装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的安装目录,因为稍后会解压到这里):
win10 rtx2060 安装cuda10.1 + cudnn7.6 + tensorflow2.0-gpu
把压缩包里红框的三个文件夹里的东西,
win10 rtx2060 安装cuda10.1 + cudnn7.6 + tensorflow2.0-gpu

导入到“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__

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

相关文章:

  • 2021-10-28
  • 2022-01-17
  • 2021-06-14
  • 2022-01-05
  • 2022-12-23
  • 2021-04-08
  • 2022-12-23
  • 2021-12-30
猜你喜欢
  • 2021-04-25
  • 2022-01-11
  • 2021-09-16
  • 2021-09-19
  • 2021-09-14
  • 2022-12-23
  • 2021-11-27
相关资源
相似解决方案