[纪念浪费的时间] 安装tensorflow-gpu + cuda + cudnn 之那些踩过的坑

首先一定要确认下载到的tensorflow cuda 和cudnn的版本是正确的

1.首先卸载tensorflow(cpu)版本,(如果有的话),没有就不用了

2.使用anaconda,安装tensorflow-gpu ,输入pip install tensorflow-gpu即可(注意,这里要架梯子才快,此外要以管理员权限打开anaconda prompt)

3.下载cuda,我在北邮人上下载的cuda9.0

4.下载cudnn,注意,一定要下载for cuda9.0的版本

Download cuDNN v7.1.3 (April 17, 2018), for CUDA 9.0

5.安装cuda,安装失败的话,多装几次,安装完毕后一定要重启,否则cuda自动卸载.........exm???

6.添加cuda 的环境变量

7.cudnn的zip解压缩后一定要放在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0下,不要放在CUDA下

8.更新gpu驱动,tensorflow最好是最新版本的1.8.0

阅读更多

没有更多推荐了,返回首页