人工智能- windows10环境,配rtx 3060ti显卡,tensorflow-gpu安装
创始人
2024-05-26 22:12:02
0

文章目录

    • 前言
    • 流程
    • 方法
    • 1.先安装网盘里的anaconda文件,安装后就是python3.8.8环境
    • 2.安装vs2019
    • 3.vs2019安装完毕后开始安装cuda
    • 4.安装cudnn
    • 5.安装tensorflow-gpu
    • 6.测试GPU是否正常识别,tensorflow是否可用

前言

最近显卡降价,入手了一块RTX3060TI战斧显卡,美滋滋,顺便升级一下深度学习的单机配置
特别说明的是30系列显卡之后必须用cuda11.0以上,因为之后的架构发生了改变,否则30系显卡无法调用
这里我用的环境是

  1. 系统:win10
  2. 显卡:RTX3060TI
  3. python:3.8.8
  4. annaconda:Anaconda3-2021.05-Windows-x86_64.exe
  5. cuda:cuda_11.0.2_451.48_win10.exe
  6. cudnn:cudnn-11.0-windows-x64-v8.0.4.30.zip
  7. VS2019
  8. tensorflow-gpu 2.4.0
    文件好多???
    阿里网盘已经准备好资料
    https://www.aliyundrive.com/s/c7bPvR9G1hF
    在这里插入图片描述
    vs2019需要下载安装器联网下载,而且组件需要自己选择,这个没法存进网盘
    tensorflow-gpu 2.4.0使用conda的pip3命令安装即可没有必要存进网盘了,可以使用清华镜像,下方有链接,其他文件都在

流程

方法

可是怎么事先知道那个版本的windows系统,显卡型号,tensorflow,cuda,cudnn是相互配合的呢?
先看显卡,看看显卡支持到哪里?
找到nvidia控制面板,看一下支持的cuda版本信息,要小于显卡所标识的最大版本号在这里插入图片描述
cmd输入nvidia-smi,查询驱动器版本号,
在这里插入图片描述
那么这里选择cuda11.0是可行的,然后找官方给的配置单,看一下相互对应的tensorflow等等
去英伟达官网,他们给的有版本配置单,按着来就行了
https://tensorflow.google.cn/install/source?hl=en#gpu
在这里插入图片描述
那么知道为什么用这些文件了,就进入文件安装使用吧

1.先安装网盘里的anaconda文件,安装后就是python3.8.8环境

在这里插入图片描述

2.安装vs2019

cuda安装需要vs环境,适配的组件在vs2019中都具备了,因此要先安装vs2019
需要注意的是在安装时勾选以下组件,否则还是没有cuda运行环境
从网上下载个vs2019安装器,然后勾选如下即可,等待安装完成
在这里插入图片描述

3.vs2019安装完毕后开始安装cuda

cuda第一开始显示的文件路径框是自解压,选择一个空白文件夹即可,安装完毕后其会将解压的文件自动删除
然后没什么可说的,全部勾选,下一步下一步
如果有人报错了,看一下安装vs2019时的两个运行环境勾选没有,或者看看你的显卡的支不支持这个cuda文件?

4.安装cudnn

其实就是解压的文件夹,将cudnn中的三个文件复制放入cuda的安装目录如下
cudnn解压后的文件夹,全选复制一下
在这里插入图片描述
粘贴到cuda安装目录
在这里插入图片描述

5.安装tensorflow-gpu

使用清华镜像更快一些
不需要清华镜像就把后面网址删掉

pip install tensorflow-gpu==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple #版本可任意修改

在这里插入图片描述

6.测试GPU是否正常识别,tensorflow是否可用

GPU是否识别

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

在这里插入图片描述

相关内容

热门资讯

中证A500ETF摩根(560... 8月22日,截止午间收盘,中证A500ETF摩根(560530)涨1.19%,报1.106元,成交额...
A500ETF易方达(1593... 8月22日,截止午间收盘,A500ETF易方达(159361)涨1.28%,报1.104元,成交额1...
何小鹏斥资约2.5亿港元增持小... 每经记者|孙磊    每经编辑|裴健如 8月21日晚间,小鹏汽车发布公告称,公司联...
中证500ETF基金(1593... 8月22日,截止午间收盘,中证500ETF基金(159337)涨0.94%,报1.509元,成交额2...
中证A500ETF华安(159... 8月22日,截止午间收盘,中证A500ETF华安(159359)涨1.15%,报1.139元,成交额...
科创AIETF(588790)... 8月22日,截止午间收盘,科创AIETF(588790)涨4.83%,报0.760元,成交额6.98...
创业板50ETF嘉实(1593... 8月22日,截止午间收盘,创业板50ETF嘉实(159373)涨2.61%,报1.296元,成交额1...
港股异动丨航空股大幅走低 中国... 港股航空股大幅下跌,其中,中国国航跌近7%表现最弱,中国东方航空跌近5%,中国南方航空跌超3%,美兰...
电网设备ETF(159326)... 8月22日,截止午间收盘,电网设备ETF(159326)跌0.25%,报1.198元,成交额409....
红利ETF国企(530880)... 8月22日,截止午间收盘,红利ETF国企(530880)跌0.67%,报1.034元,成交额29.0...