tensorflow-gpu 2.1.0安装教程及问题注意
写在前面
本文主要是针对win10系统下的tensorflow gpu 版本的下载安装途径,在安装过程成,也遇到了诸多问题,故记录下来,一方面是为了作为备份,供自己以后查阅使用,另一方也是希望下文中提到问题,如果你也遇到,可以提供解决思路和方法,针对文中的提出的问题、方法、步骤,如有疑问或建议,也欢迎与我分享。
基本环境配置情况
python:3.6.x
cuda:10.1
cudnn:7.6.5
numpy:1.19.2
1 Anaconda 下载
Anaconda 下载安装这里不做赘述,网络上资源很多,主要提提醒安装完anaconda 后检查系统环境变量中是否添加了anaconda的变量,另外一个就是下载源换成清华的下载源
2 CUDA和cuDnn下载安装
我看过的很多教程都会先介绍如何下载cuda和cudnn,我也不能免俗,但是在实际安装中,我非常强烈建议你先跳过这一节,不要手动安装cuda和cudnn,而是尝试先在anaconda新建的虚拟环境中下载安装tensorflow-gpu,观察conda是否会自动下载cudatoolkit 和 cudnn,如果conda自动下载的话可以省去很多工作
版本说明
cuda和cudnn的版本主要是考虑你要是用的tensoflow的版本,需要与之兼容。下图来源于tensorflow官网,