tensorflowGPU安装+经验分享

历时两天，踩过很多坑，终于语气词装好了。

说一下我的情况：tensorflowGPU-1.14.0，CUDA-10.0，cuDNN-v7.6.5，Anaconda3-2019.10，python-3.6，1650显卡。

2020年3月2日

 

好了下面是步骤！

大致的步骤为

一、安装CUDA和cuDNN。

二、安装Anaconda

三、安装tensorflowGPU

 

这里介绍几篇比较详细的文章：

https://zhuanlan.zhihu.com/p/37086409

https://blog.csdn.net/weixin_44170512/article/details/103990592

下面一一介绍：

一、安装CUDA和cuDNN。

确认版本，这里有两个说法：

https://github.com/fo40225/tensorflow-windows-wheel这里的README文件有一种版本说法

https://tensorflow.google.cn/install/source_windows这里是google官方的说法

CUDA下载地址https://developer.nvidia.com/cuda-toolkit-archive

cuDNN下载地址https://developer.nvidia.com/rdp/cudnn-download

（如果链接失效请自行搜索CUDA和cuDNN）

CUDA直接安装，cuDNN需要将对应的三个文件拷入相应的CUDA目录下。

cuDNN中的----------->

我尝试CUDA10.1+tensorflowGPU2.1和2.0+python3.7都失败了，错误提示：找不到指定模块。

后来我的做法是先安装先安装的tensorflowGPU版本，然后根据报错信息来选择CUDA，然后根据CUDA选择cuDNN。

根据错误信息ImportError: Could not find 'cudart64_100.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 10.0 from this URL: CUDA Toolkit 9.0 Downloads

所以根据提示下载CUDA10.0

下载cuDNN时，需要注意选择最新的，不然会出现cuDNN failed to initialize错误。（具体错误信息不记得了，只记得关键词）

二、安装anaconda

不需要太多说明， 官网下载直接安装就好。

三、安装tensorflowGPU

进入Anaconda prompt。新建rensorflow环境。

conda create -n tensorflow pip python=3.6

（这条语句的python=3.6可以指定其他版本）

激活该环境

activate tensorflow

进入tensorflow环境后前面会提示。

这个时候安装想安装的GPU版本。

pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

这里的可以指定GPU版本，否则默认安装最新的，我在安装时默认最新2.1  可以在tensorflow-gpu后面加==1.14.0（指定版本）

（后面清华的镜像快的亚批）

到此就安装完成了，很简单，下面是一些分享。

prompt的一些命令

查看目前有哪些环境：（也可以查看Anaconda里的envs文件夹）

conda info --envs

查看当前环境下安装的包：

pip list

卸载某版本tensorflowGPU：（版本自行更改，该方式还可以卸载其他库）

pip uninstall tensorflow-gpu==1.14.0

删除tensorflow环境：（名称和当初新建的环境一致）

conda remove -n tensorflow --all

完全卸载Anaconda：参考https://blog.csdn.net/qq_40605139/article/details/95249601

 

关于环境变量

安装Anaconda时需要配置。可以参考https://www.jianshu.com/p/d3a5ec1d9a08

CUDA和cuDNN不需要配置，安装时自动配置的。

如果有其他问题请留言，如果我有知道的会及时评论的。

如果什么地方有错误请斧正。