【002】配置深度学习GPU框架(Win10TensorFlow-GPU框架+pytorch-GPU框架)

说在前面的话:一定要记住,安装前一定要确认好每个软件的版本是否相互支持
GPU加速环境——见博客:
【001】配置深度学习GPU加速环境(win10系统+GPU2080 Ti+NVIDA显卡驱动+CUNA+cuDNN) - BG大龍的博客

参考文献:
1—— CUDA各版本与驱动的对应关系以及tensorflow 各个版本与python\CUDA\cudnn版本对应关系 - 页页读
2—— Tensorflow不同版本要求与CUDA及CUDNN版本对应关系 - LoveWeeknd https://blog.csdn.net/oMoDao1/article/details/83241074
3——安装tensorflow GPU版本–tensorflow-gpu版本与CUDA版本对应关系 - sunmingyang1987的博客 https://blog.csdn.net/sunmingyang1987/article/details/102872658

1、‘版本相互支持’的重要性

在这里插入图片描述

1.1——这是CUDA包和显卡驱动的【适配情况】
在这里插入图片描述
1.2——这是TensorFlow各版本的适配情况【Windows端】
链接: 在 Windows 环境中从源代码编译 | TensorFlow

这是CPU情况,在这里插入图片描述
这是GPU情况,
在这里插入图片描述
另外需要注意,在一个文章中看到,
在这里插入图片描述

2、实验环境

【1】
操作系统win10、
Anaconda3.5.1、
Python:可以创建独立环境,一定要适配框架对应的
编译器:Pycharm2019.2

【2】
GPU显卡:NVIDIA RTX 2080 Ti
显卡驱动:441.66版,要和cuda版本适配
CUDA:10.1
cuDNN:7.6

【3】 框架
TensorFlow-gpu,
pytorch-gpu
创建独立环境,
注意版本,和硬件软件之间的兼容关系

TensorFlow-GPU框架

1、TensorFlow虚拟环境

pytorch-GPU框架

下载
官网链接:https://pytorch.org/get-started/locally/
选择适配的,
在这里插入图片描述

安装
在anaconda中创建虚拟环境——pip或者conda指令安装——将pytorch配置到pycharm中

测试

Anaconda Prompt——conda activate pytorch——python——import torch
—— torch.cuda.is_available()
在这里插入图片描述
返回结果为True,则表示GPU可用

参考文献:
Win10 系统在 Visual Studio 2019 环境下配置 CUDA 10.1 + TensorFlow-GPU 1.14.0 - 半吊子程序员的技术专栏 https://blog.csdn.net/u012319441/article/details/95551632

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值