tensorflow2.1.0+cuda10.1+cudnn7.6安装

1.安装anaconda,python3.7版本
链接:https://repo.anaconda.com/archive/Anaconda3-2020.02-Windows-x86_64.exe
注意勾选添加环境变量,其他默认安装
2.打开anaconda prompt(Anaconda3)创建环境

conda create -n TF2.1 python=3.7

目前直接这样创建环境会报错,原因是清华镜像不能用了,所以需要删除镜像链接:

conda config --remove-key channels

再次创建即可成功
3.安装cuda10.1和cudnn7.6

conda install cudatoolkit=10.1
conda install cudnn=7.6

4.安装tensorflow2.1.0

pip install tensorflow==2.1

2.1.0默认安装GPU版本,这个安装较慢,容易安装失败,可用豆瓣的镜像:

pip install tensorflow==2.1.0 -i https://pypi.douban.com/simple

5.验证安装是否成功

python
import tensorflow as tf

如果没有报错则安装成功,如果报错DLL load failed: 找不到指定的模块
原因:可能是Tensorflow 2.1.0 运行需要安装 VS 2019 的环境,需要下载相关环境运行包
链接:https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads
注意下载x64版本,然后安装,再次import成功,查看tensorflow版本,显示2.1.0则表示安装成功
在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值