历时两天,踩过很多坑,终于语气词装好了。
说一下我的情况: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
下面一一介绍:
一、安装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://www.jianshu.com/p/d3a5ec1d9a08
CUDA和cuDNN不需要配置,安装时自动配置的。