【Python】Windows10+RTX-3080+安装Tensorflow2-gpu版本成功安装指南(2020.12更新)

前言

台式机换了个RTX-3080的显卡,要重新装机并配置tensorflow的环境,踩了超多的坑,现在记录一下成功的经历,真的太不容易了…我系统用的Windows,网上大多数人用的都是Linux系统,但是我不想再花时间去学一个系统了。网上说30系列显卡只支持cuda11及以上,这里也不尝试cuda10了。而且我之前自己搭网络用的tf2.0,本文也不尝试安装tf1.0的版本了。先列出我最后成功的版本:

  • 系统:Windows10
  • 显卡:RTX-3080
  • 驱动:456.43
  • cuda:CUDA11.1.0_win10_network
  • cudnn: cudnn-11.1-windows-x64-v8.0.5.39
  • tensorflow:tf-nightly-gpu 2.5.0.dev20201217

一、安装Anaconda

打开清华镜像站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D

我下了个最新的版本:Anaconda3-2020.11-Windows-x86_64.exe

安装过程,我这次做出的改动有:

1.安装Install for:“All Users”

2.直接自动添加环境变量了,即勾选了“Add Anaconda3 to the system PATH environment variable”

二、查看驱动

RTX3080采用了新的Ampere架构GA102-200,那么显卡驱动也必然是需要最新的,可以参考cuda与驱动对应的关系

在这里插入图片描述

想要装CUDA 11.1的话,驱动版本要在456.38以上。想要看自己的驱动版本的话,可以在cmd中输入:

nvidia-smi

可以看到驱动版本是456.55,还显示了CUDA版本为11.1。

在这里插入图片描述

也可以不提前安装驱动,CUDA11安装里面自带了驱动,因为我后面卸载重装了,就直接用的CUDA11安装的驱动(自动安装的版本为456.43),如下所示:

在这里插入图片描述

三、安装CUDA

  1. 到链接: 英伟达官网下载对应的CUDA版本。
    我下载的版本是CUDA11.1.0 ,选择"Windows;x86_64;10;exe(network)"

  2. 安装CUDA:双击执行下载的exe文件,会先解压文件到临时目录(不是安装目录),保持默认即可.【我不知道可不可以下到别的里面,就照着博客下了】

在这里插入图片描述

  1. 这里需要选择自定义安装:

在这里插入图片描述

  1. 自定义安装选项,不勾选“Visual Studio Integration”,勾选的话后面会有个什么提示,所以又退回这步不勾选了。

在这里插入图片描述

  1. 这三个地址十分重要,后续我们将进行环境变量的配置,不要改变路径。

在这里插入图片描述

  1. 之后等待安装,遇到了下面的情况

在这里插入图片描述

Installed:
     - Nsight Monitor
Not Installed:
     - Nsight for Visual Studio 2019
       Reason: VS2019 was not found
     - Nsight for Visual Studio 2017
       Reason: VS2017 was not found
     - Nsight for Visual Studio 2015
       Reason: VS2015 was not found
     - Integrated Graphics Frame Debugger and Profiler
       Reason: see https://developer.nvidia.com/nsight-vstools
     - Integrated CUDA Profilers
       Reason: see https://developer.nvidia.com/nsight-vstools

不知道有没有什么问题,先继续往下吧。

  1. 配置CUDA环境变量

网上很多说要环境变量>系统变量>Path中手动添加路径的,我整合了好多条:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin【自动配的】
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\libnvvp【自动配的】
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolk

  • 16
    点赞
  • 55
    收藏
    觉得还不错? 一键收藏
  • 28
    评论
### 回答1: 要在RTX 305上安装TensorFlow GPU,您需要执行以下步骤: 1. 安装CUDA Toolkit:访问NVIDIA官方网站,下载并安装与您的操作系统和显卡型号兼容的CUDA Toolkit。 2. 安装cuDNN:访问NVIDIA官方网站,下载并安装与您的CUDA版本兼容的cuDNN。 3. 安装TensorFlow GPU:使用pip命令安装TensorFlow GPU。例如,输入以下命令:pip install tensorflow-gpu 4. 验证安装:启动Python解释器,导入TensorFlow并运行一些TensorFlow代码,以确保TensorFlow GPU已正确安装并可以正常工作。 请注意,安装TensorFlow GPU需要一些计算机知识和技能。如果您不确定如何执行上述步骤,请寻求专业人士的帮助。 ### 回答2: RTX3050是一款性能出色的显卡,可以用于深度学习任务。而tensorflow-gpu则是一种广泛使用的深度学习框架,可以提高计算速度和效率。在安装tensorflow-gpu的过程中,需要注意以下几点。 第一,确保显卡驱动程序已正确安装。RTX3050显卡需要安装最新的驱动程序,以确保其正常运行并提供最佳性能。可以从官方网站或显卡厂商网站下载并安装驱动程序。 第二,安装CUDA。CUDA是一种用于GPU加速的计算平台,可大大提高计算速度。可以从 NVIDIA 的官方网站上下载和安装适用于 RTX3050 显卡的最新 CUDA 版本。 第三,安装cuDNN。cuDNN是一种用于加速深度神经网络训练的库。可以从 NVIDIA 的官方网站上下载适用于RTX3050显卡的最新版本。 第四,下载和安装TensorFlow-GPU。在安装tensorflow-gpu之前,需要确保安装了适用于RTX3050 GPU的CUDA和cuDNN。可以使用包管理器,如pip,进行安装。首先要确保pip已经安装。然后可以使用pip install tensorflow-gpu 命令来安装tensorflow-gpu。 第五,检查安装是否成功。可以在Python中导入tensorflow-gpu并运行简单的深度学习代码,以确认安装是否成功。如果能正常运行,说明tensorflow-gpu已经成功安装。 总之,安装tensorflow-gpu需要正确安装相应的显卡驱动程序、CUDA和cuDNN,并使用包管理器pip进行安装成功安装后,可以使用导入tensorflow-gpu并运行简单示例来检查。 ### 回答3: RTX 3050是一款新款显卡,其GPU运算速度非常快,可用于机器学习和深度学习等高性能计算领域。 在安装TensorFlow-GPU时,首先需要确保显卡已经成功安装了CUDA和cuDNN。使用RTX 3050,可以选择CUDA 11.1版本和cuDNN 8.0版本。需要注意的是,安装支持RTX 3050的CUDA和cuDNN版本是至关重要的。 接下来,用户需要下载TensorFlow-GPU的whl文件。TensorFlow官网提供了不同的whl文件,可以根据用户的操作系统选择下载。如果使用Windows操作系统,则需要下载TensorFlow-GPU whl文件,文件名中包含了CUDA和cuDNN版本号信息,这是为了保证TensorFlow和CUDA及cuDNN的版本相匹配。 在下载完whl文件后,用户可以通过pip install命令进行安装,具体的命令为:pip install tensorflow-GPU-[version]-cp[python版本]-cp[python版本]m-win_amd64.whl,其中version是TensorFlow版本号,[python版本]是当前系统上所使用的Python版本号。 如果在安装过程中遇到问题,可以通过查看TensorFlow官方文档或论坛上其他开发者发布的解决方案来解决问题。此外,也可以通过与其他开发者的交流或咨询专业技术人员来解决问题。 总的来说,安装TensorFlow-GPU需要注意CUDA和cuDNN版本号的选择,以及TensorFlow版本Python版本的匹配。只要按照正确的步骤操作,就能成功安装TensorFlow-GPU并开始进行高性能计算和深度学习应用。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值