Anaconda下conda,pip安装tensorflow-gpu

conda安装tensorflow-gpu

可以使用conda命令傻瓜式安装tensorflow-gpu,只需要一个命令就可以自动下载安装。

conda install tensorflow-gpu

但是19年之后,清华等国内镜像不再支持conda安装tensorflow-gpu,因此导致有时会下载失败,我就曾经尝试了清华,中科大,豆瓣,阿里云的镜像都没有完成安装,因此如果conda安装失败可以采取原来的笨方法,用pip分三步安装tensorflow-gpu。

pip安装tensorflow-gpu

pip安装tensorflow主要分为三步:
1. 安装CUDA
2. 安装cuDNN
3. 安装tensorflow

一、安装CUDA
1.在下面这个网址下载适合自己版本的CUDA(我的是py3.6+CUDA9.0+tensorflow1.12.0)。
https://developer.nvidia.com/cuda-downloads

还可以点击Legacy Releases选择之前比较老的版本。

2.下载好之后直接安装选择默认路径即可。
3.将CUDA安装目录下的bin和lib\x64添加到Path环境变量中。

二、安装cuDNN
1.在下面这个网址下载适合CUDA版本的cuDNN。
https://developer.nvidia.com/cudnn
2.下载完成之后解压。
3.将压缩包中的bin,include,lib分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0下,直接选择覆盖文件即可。(这是CUDA的安装目录,我下载的是9.0版本,所以是v9.0,以下类同。)
4.将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64中的cupti64_90.dll复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin目录下即可。

三、安装tensorflow-gpu
打开Anaconda Prompt,输入命令:

pip install tensorflow-gpu

这个默认的是从国外源下载,速度较慢,也可以指定为国内镜像,下面是以清华镜像为例

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

到这里就大功告成了,可以在使用import tensorflow as tf 来验证有没有安装成功。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值