Win10下Tensorflow2.1环境的配置

记录自己配置tensorflow2.1的过程

1.安装Anaconda

官方下载地址:https://www.anaconda.com/products/individual
自行选择64位或32位
Win10下Tensorflow2.1环境的配置_第1张图片
安装过程中注意勾选添加环境变量(Add Anaconda to the system PATH environment variable)

2.安装CUDA和cudnn

Tensorflow官方要求:
Win10下Tensorflow2.1环境的配置_第2张图片
首先确认电脑的显卡品牌为NVIDIA,cmd中输入nvidia-smi查看驱动版本,tensorflow2.1需要418.X或更高版本。
Win10下Tensorflow2.1环境的配置_第3张图片
安装CUDA10.1和cudnn7.6.5:
CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive
(下载CUDA Toolkit 10.1 update2版本)
cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-download
(选择Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.1
其中cudnn是压缩包,解压后放在cuda安装目录CUDA/v10.1/下(放在一起添加环境变量时方便)。
最后,设置下环境变量即可。
Win10下Tensorflow2.1环境的配置_第4张图片
Win10下Tensorflow2.1环境的配置_第5张图片

3. 安装Tensorflow2.1

pip install tensorflow-gpu==2.1.0

速度慢加上豆瓣源:-i https://pypi.douban.com/simple/
测试:
Win10下Tensorflow2.1环境的配置_第6张图片

4. 可能出现的问题:import tensorflow出错

安装vc++2015-2019相关库:
链接:https://pan.baidu.com/s/1cJsW98MaIR0vCL1OP4uynQ
提取码:snui

你可能感兴趣的:(Deep,Learning)