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

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

2020年3月2日

 

Tensorflow使用GPU加速的教程 tensorflow gpu_python

Tensorflow使用GPU加速的教程 tensorflow gpu_python_02

好了下面是步骤!

大致的步骤为

一、安装CUDA和cuDNN。

二、安装Anaconda

三、安装tensorflowGPU

 


下面一一介绍:

一、安装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中的

Tensorflow使用GPU加速的教程 tensorflow gpu_tensorflow_03

----------->

Tensorflow使用GPU加速的教程 tensorflow gpu_tensorflow_04

我尝试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环境后前面会提示。

Tensorflow使用GPU加速的教程 tensorflow gpu_python_05

这个时候安装想安装的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

Tensorflow使用GPU加速的教程 tensorflow gpu_CUDA_06

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

pip list

卸载某版本tensorflowGPU:(版本自行更改,该方式还可以卸载其他库)

pip uninstall tensorflow-gpu==1.14.0

删除tensorflow环境:(名称和当初新建的环境一致)

conda remove -n tensorflow --all


 

关于环境变量

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

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

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