windows10 环境下配置tensorflow-gpu 环境

显卡型号支持查询:https://developer.nvidia.com/cuda-gpus

Tensorflow不同版本要求与CUDA及CUDNN版本对应关系:https://blog.csdn.net/omodao1/article/details/83241074

下载cuda,官网下载即可。 目前这种方式是最靠谱的。https://developer.nvidia.com/cuda-downloads

安装cuDNN下面是下载地址,需要提前注册。 注册一下就好。注意下载的版本。https://developer.nvidia.com/rdp/cudnn-download

Copy CuDNN files to Nvidia CUDA toolkit folder when 2. has completed (usually is located on C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)

  • copy cudnn\bin\cudnn64_5.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\
  • copy cudnn\include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include\
  • copy cudnn\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64\

在PATH中,添加 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

以及解压出的cudnn文件中bin的目录C:\tools\cuda\bin

  • download and install these packages via pip (move to the directory where they were downloaded, open a command prompt and make 'pip install package_name')
  • open command prompt and make 'pip install matplotlib'
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值