XP2020的个人博客分享 http://blog.sciencenet.cn/u/XP2020

博文

Tensorflow2.1安装过程记录

已有 3527 次阅读 2020-5-23 18:04 |个人分类:AI|系统分类:科研笔记

过程参考:https://www.bilibili.com/video/BV1UE411N7gS/?spm_id_from=333.788.videocard.1

 

软硬件要求:https://www.bilibili.com/video/BV1Zt411T7zE?p=2

 

系统:

64bit: ubuntu 16.04及以上,windows7及以上(仅支持python3),Raspbian9.0及以上

pip 要求不低于19.0

 

GPU:

检查GPU品牌及其驱动版本:win+R: dxdiag

算力不低于3.0: https://developer.nvidia.com/cuda-gpus

驱动不低于410.x ,使用命令 nvidia-smi

 

详细步骤:

1.     下载anaconda,安装(勾选添加环境变量),Python3.7版本,与系统匹配。

2.     打开anaconda prompt,新建虚拟环境,

(base) C:\Users\admin>conda create -n TF2 python=3.7

3.     激活虚拟环境

(base) C:\Users\admin>conda activate TF2

4.     安装cuda  (cudatoolkit-10.1.243 | 300.3 MB)

(TF2) C:\Users\admin>conda install cudatoolkit=10.1

5.     安装cudnn(cudnn-7.6.5 | 179.1 MB  |)

(TF2) C:\Users\admin>conda install cudnn=7.6

6.     安装tensorflow (tensorflow 2.1.0)使用豆瓣源加速下载,tensorflow-2.1.0-cp37-cp37m-win_amd64.whl (355.8 MB)

(TF2) C:\Users\admin>pip install tensorflow==2.1 -i https://pypi.douban.com/simple

7.     检查是否安装成功。

(TF2) C:\Users\admin>python

>>> import tensorflow as tf

报错:ImportError: DLL load failed: 找不到指定的模块。

参考如下链接,安装VC_redist.x64.exehttps://blog.csdn.net/lingtianyulong/article/details/104207960?utm_medium=distribute.pc_relevant_right.none-task-blog-BlogCommendFromMachineLearnPai2-3.nonecase&depth_1-utm_source=distribute.pc_relevant_right.none-task-blog-BlogCommendFromMachineLearnPai2-3.nonecase

 

下载安装后在运行上述两条命令,输出正常:

(TF2) C:\Users\admin>python

>>> import tensorflow as tf

2020-05-23 17:52:09.086574: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll

 

>>> tf.__version__

'2.1.0'

>>> 

 

安装完成。




https://blog.sciencenet.cn/blog-3426187-1234600.html

上一篇:[转载] A Survival Guide to a PhD by Andrej Karpath
收藏 IP: 39.82.249.*| 热度|

0

该博文允许注册用户评论 请点击登录 评论 (0 个评论)

数据加载中...
扫一扫,分享此博文

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2024-4-19 16:09

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部