配置tensorflow-gpu和tensorflow-cpu环境和对应的keras ,pytorch(详细步骤)
前言
这只是安装对应版本的tensorflow-gpu和tensorflow-cpu环境和对应的keras ,pytorc的步骤
0. 安装Anaconda 和 微软官网的 VC++ 补丁
(1)安装Anaconda
进入官网,滑到最下面,点击安装。(安装路径可以自定义,其他都默认安装)
https://www.anaconda.com/products/individual

(2)安装VC++ 补丁
到微软官网下载对应VC++ 补丁并安装。
https://visualstudio.microsoft.com/zh-hans/downloads/

1.查看版本对应关系
(1)tensorflow与keras版本匹配情况
网址: https://docs.floydhub.com/guides/environments/
目前最新支持keras 的tensorflow 版本为 2.2.0 (建议下载这个版本)

(2)tensorflow与CUDA版本以及cuDNN匹配情况
网址:https://tensorflow.google.cn/install/source_windows?hl=cn
tensorflow2.2.0 对应的cuDNN7.6, CUDA10.1

(3)CUDA与pytorch版本对应情况
查看pytorch官网:
https://pytorch.org/get-started/previous-versions/

(4)最终下载的各个版本
python 3.8 (anaconda自带)、cuDNN 7.6、 CUDA 10
一站式教程:配置TensorFlow-GPU/CPU与Keras、PyTorch环境

最低0.47元/天 解锁文章
3350

被折叠的 条评论
为什么被折叠?



