||
过程参考: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: 找不到指定的模块。
下载安装后在运行上述两条命令,输出正常:
(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'
>>>
安装完成。
Archiver|手机版|科学网 ( 京ICP备07017567号-12 )
GMT+8, 2024-4-19 16:09
Powered by ScienceNet.cn
Copyright © 2007- 中国科学报社