最新tensorflow安装

本文详细介绍了在Windows10上,使用Anaconda3、Python3.7和PyCharm安装带有GPU支持的Tensorflow 2.1的步骤,包括CUDA 10.2和cuDNN的安装,以及解决 cudart64_101.dll 错误的方法。
摘要由CSDN通过智能技术生成

windows10+anaconda3+python3.7+pycharm+gpu–tensorflow 2.1

NVIDIA GeForce GTX 1650版本441.22安装的CUDA版本是10.2.9

CUDA

有关显卡,显卡驱动,cuda ,cudnn是什么,以及之间的关系可自行搜索

在左下角搜索框搜索NVIDIA进入控制面板,查询当前版本在这里插入图片描述
算力需大于3.1

查询CUDA版本和显卡驱动版本 需按对应下载

CUDA下载 下载非常慢 😢,安装就是一直无脑下一步

测试win+R: nvcc -V

在这里插入图片描述

出现上图证明成功。

cuDNN

cuDNN下载 直接下载要注册,右键选择复制链接,在打开迅雷即可,或者直接右键迅雷下载,我选择的是for cuda9.0的版本,没选择最新的,好像问题不大,也有可能比较慢,

最后得到一个zip文件,解压后得到三个文件夹,如下所示:
在这里插入图片描述

将bin ,include 和lib文件夹的内容分别复制到cuda安装路径下与cuda的bin ,include 和lib文件夹合并。

测试安装是否成功

win+R :cmd

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite

.\deviceQuery.exe显示如下结果说明安装成功:
在这里插入图片描述

Tensorflow

win+R

python -m pip install --upgrade pip
pip install tensorflow-gpu==2.1

首先更新pip,第二句容易下载超时,不行则换成下面这句

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow 

指定镜像源会比较快,安装需要一点时间

pycharm

pycharm安装

新建一个项目文件,将anaconda和pycharm绑定

file->settings->project interpreter ->右边小齿轮->show all ->+号->一般在c盘user路径下选择anaconda文件中的Python.exe

新建一个python程序

import tensorflow as tf
tf.__version__

首次运行出现如下错误

Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found

Ignore above cudart dlerror if you do not have a GPU set up on your machine.

通过这篇博客顺利解决

先来排查错误,显示“cudart64_101.dll not found”,那是因为我安装的cuda为10.0版本的,“cudart64_101.dll”是cuda10.1这个版本中才含有的(cuda10.2貌似也没有)。

因此我们需要手动将“cudart64_101.dll”添加到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin”中去。

另附cudart64_101.dll链接

再次运行成功

在这里插入图片描述
希望对你有所帮助!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值