Windows10 安装tensorflow2.0.0

Windows10 安装tensorflow2.0.0

安装Anaconda

直接官网下载好安装就好了。

安装CUDA

我装的是10.0的,好像安装tensorflow2.0.0只能装CUDA10.0(或者说安装了CUDA10.0只能装tensorflow2.0.0),因为后面运行的时候需要加载cudart64_100.dll文件等,只有在CUDA10.0的bin目录下才有这个文件,10.1版本的就是cudart64_101.dll了。安装过程简单介绍下:
在安装之前,先看一下是不是之前已经装过了CUDA,如果装了版本不一样的,就先卸载之前的,卸载方法是在控制面板中,带NVIDIA的软件,只保留“图形驱动程序”、“Physx”和“GeForce Experience”,其他都卸载。
处理好之前的,然后去官网找到CUDA10.0下载.exe文件,下载好之后直接双击就可以安装了。

装的时候选择自定义安装。

画红框的那个不用选。


下图,如果右侧版本号大于左侧,一定不要打钩。

安装时下面图片中的路径可以自己改,我是直接装到D盘的,前两个放在同一个文件夹(CUDA1),最后一个分开放(CUDA2),反正都在一个总文件夹中。

这样之后直接下一步就可以继续安装了。
完成之后在cmd中断中运行nvcc -V,如果显示CUDA版本号,就安装成功了。

安装Cudnn

还是去官网下载,这个需要注册账号,我用的是谷歌邮箱(gmail.com),直接在手机上下载QQ邮箱客户端,登录谷歌邮箱可以接收邮件。

一定要看准和CUDA10.0对应的版本。
下好之后解压,记得改下文件名为cudnn,直接放到CUDA安装路径最初始的路径下就行,就是和上面的CUDA1并列的。

配置环境变量



添加4个环境变量,把这四个都移到最上边,提高搜索效率。

安装tensorflow2.0.0

先用anaconda创建python3.6的虚拟环境:
输入如下命令

// 创建虚拟环境
conda create -n tf20 python=3.6
//进入anaconda
activate
//进入虚拟环境
conda activate tf20
//安装tensorflow2.0.0,使用豆瓣源比较快
pip install tensorflow_gpu==2.0.0 -i https://pypi.douban.com/simple --trusted-host pypi.douban.com

虚拟环境改名字:

// An highlighted block
conda create --name python32(新的名字) --clone python321(老名字)
conda env remove --name old_name

安装时可能出现的问题

【anaconda】Collecting package metadata (current_repodata.json): failed / Solving environment: failed
解决方法:按以下操作将文件粘贴过去
原来在Anaconda3/Library/bin中,粘贴到Anaconda3/DLLs中

libcrypto-1_1-x64.dll
libssl-1_1-x64.dll

或者更换镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

测试是否安装成功

//在虚拟环境下,进入python环境
python
import tensorflow as tf
tf.test.is_gpu_available()

输出True就大功告成了!

参考文献:

https://www.bilibili.com/video/BV1HV411q7xD
https://www.bilibili.com/video/BV1Lv41177BW
https://blog.csdn.net/miracleoa/article/details/106039388

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值