您的位置:首页 > 其它

最新tensorflow安装

2020-06-03 05:41 295 查看

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链接

再次运行成功


希望对你有所帮助!

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: