Windows10下Anaconda安装tensorflow2.1.0

Windows10下Anaconda安装tensorflow2.1.0

笔者最近在安装tensorflow2.1.0,有一些心得,仅供大家参考。

首先是电脑环境,笔者电脑有
Anaconda3\Pycharm\Visual Studio 2019\CUDA 10.1\cuDNN10.1

Anaconda官网
https://www.anaconda.com/distribution
安装方法有很多可以自行搜索,我的建议就是安装步骤来就行,记住自己安装路径,安装完后,将目录添加到系统环境变量path就可以了。

Pycharm官网
https://www.jetbrains.com/pycharm/download/#section=windows
同样有很多安装教程,我的建议也是安装步骤来就行。

目前直接PIP安装最新tensorflow会直接GPU版本故先装CUDA和cuDNN

tensorflow2.1.0对应的是CUDA10.1,如果版本不对,可能会出现找不到cudart64_101.dll的情况

CUDA官网下载,选择Windows 64位系统 win10 local本地下载,
https://developer.nvidia.com/cuda-toolkit-archive
选择CUDA10.1

cuDNN官网下载,不过需要提前注册NVIDIA账号才行,选择对应的10.1下载
https://developer.nvidia.com/rdp/cudnn-download
选择对应的10.1下载
双击运行cuda_10.1.243_426.00_win10,建议都默认设置来就可以了,一般默认系统环境变量path也会在安装中自动添加。
笔者电脑已经有visual studio,所以没有的人,可以在安装过程中点高级选项,把Visual Studio(VS) Intergration√取消。不过安装一个visual studio也并不麻烦,可以提前安装一个Visual Studio2017或者2019的社区版,反正免费。

解压cuDNN的压缩包,将cudnn-10.1-windows10-x64-v7.6.5.32\cuda下的三个文件夹bin\include\lib放入CUDA安装目录即安装完成,如果是默认,就在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1。

粘贴至此
打开cmd,默认是在C:\Users\Administrator下,连续两次cd …/转到C:\然后cd转到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin,再输入nvcc -V,显示编号即成功。
编号
接下来安装tensorflow

打开Anaconda Prompt,输入如下代码

conda create -n tensorflow python=3.7
activate tensorflow
pip install --upgrade --ignore-installed tensorflow -i https://pypi.doubanio.com/simple

这里用的是豆瓣镜像源,个人觉得最快的一个了。

安装完成后,打开Anaconda Prompt输入测试代码

activate tensorflow
python

import tensorflow as tf
print(tf.__version__)
print('GPU', tf.test.is_gpu_available())
a = tf.constant(2.0)
b = tf.constant(4.0)
print(a + b)

结果如下

运行结果

这是Anaconda Navigator里面显示已安装环境:Navigator

这是Pycharm里面导入Anaconda tensorflow环境后测试:
导入

结果

完成!

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值