从零开始搭建GPU深度学习环境(pytorch)

news/2024/10/4 20:07:33/文章来源:https://blog.csdn.net/u010321827/article/details/142024570

傻乎乎的我,突然发现我自己的笔记本电脑居然有gpu,这个电脑是我弟在2017年购入的。

电脑已经按照了cpu环境,现在增加gpu环境

参考torch的cpu版本和gpu版本有什么区别 torch与cuda版本_mob64ca13f6035c的技术博客_51CTO博客


前言:
配置环境:

Windows10 + Anaconda + python3.8.10+GPU:RTX2060 ,已安装好显卡驱动
本文安装的Torch和CUDA版本:

Cuda11.1.1+cuDNN8.05+torch1.8.1+torchvision0.10.0
建议:

先了解显卡的cuda版本和cudatoolkit的区别;文中的第一项
1 显卡驱动带的Cuda和自己装的Cuda有什么区别?
疑问: 安装好CUDA后,在cmd中输入nvidia-smi和nvcc -V显示的cuda版本不同?如图所示。

2 Cuda11+CudNN安装流程
总结以下下载地址:
1.Nvidia显卡驱动:  https://www.nvidia.cn/Download/index.aspx?lang=cn2.Cuda下载:

官网最新版本: https://developer.nvidia.com/cuda-downloads
官网往期版本:  https://developer.nvidia.com/cuda-toolkit-archive
3.cuDNN下载:  https://developer.nvidia.com/rdp/cudnn-archive4.pytorch和torchvision下载:

官网最新版本:  https://pytorch.org/get-started/locally/
往期版本下载:  https://download.pytorch.org/whl/torch_stable.html
3.1 前言:
若显卡已安装好Nvidia显卡驱动,请在cmd使用nvidia-smi查看显支持的cuda的最高版本,如图所示。

若还未安装显卡驱动(也不用担心),安装cuda时勾选显卡驱动可顺便将显卡驱动一起安装

作者的显卡驱动在 https://www.nvidia.cn/Download/index.aspx?lang=cn下载并安装完毕。

3.2 Cuda安装:
前言
在3.1中nvidia-smi查询到Cuda版本为11.3(Driver API),因此在下载Cuda为深度学习加速时,只要CUDA Toolkit 的版本<=11.3。

安装步骤
步骤1: 选择合适的cuda版本,并安装,本文选择的是 11.1,如图所示。

最新版本cuda: https://developer.nvidia.com/cuda-downloads(最新版本的cuda)

往期版本cuda:https://developer.nvidia.com/cuda-toolkit-archive (很多版本任君选择)

步骤2:(安装) 双击exe文件完成解压(解压路径最好不要改),进入安装后,选择自定义安装,如下图所示。

步骤3:自定义安装说明, 选好并点击安装。

情况1:若电脑已经安装好显卡驱动,只勾选Cuda。其它组件请对比版本号。若当前版本较新,则不安装其它选项,若版本较旧,建议勾选安装,如左图所示 。
情况2:若电脑未安装显卡驱动,建议都勾选,会顺便把显卡驱动装上,如右图所示
选完后,下一步----安装即可。
步骤4: cmd中,输入nvcc -V(注意V是大写),可查当前cuda版本,即安装成功,如下图所示。

TIPS:若nvcc -V没反应,则看是否cuda未加入环境变量

1.打开高级系统设置,在用户变量的Path中加入cuda的bin和lib路径,如

3.3 cuDNN安装步骤(cuDNN8.0.5)
cuDNN下载官网 https://developer.nvidia.com/rdp/cudnn-archive 官网已说明Cuda对应的cuDNN版本,相信官网,如图所示。本文选择cuDNN8.0.5

在官网下载cuDNN前需注册Nvidia会员,免费的。

具体步骤: 下载cuDNN,并将zip文件解压,将里面的文件全部复制到CUDA的安装目录中,如图所示。

此时cuDNN也配置完成了!

4 pytorch和torchvision的安装(gpu版本,python3.8+pytorch1.8.1+torchvision0.9.1)


方法一:根据官网指令进行安装(不推荐,容易失败,因为国情原因)
网站:https://pytorch.org/get-started/previous-versions/

说明:

命令行中cudatoolkit实际上就是第3点中的Cuda(Runtime API),若已根据第3点安装了Cuda,就不需要cudatoolkit了;
第3点中的cuda是系统级安装,命令行的conda install cudatoolkit是虚拟环境级的安装
用命令行安装一般都会失败,因为网络问题和安装包比较大,因此请看方法二,用whl安装。
方法二:使用whl安装。


安装步骤:
步骤1:选择合适的torch和torchvision版本

注意:根据你实际使用的python版本来选择,cp38为python3.8,cp36为python3.6,不要选错了。

打开这个网页

步骤二: 安装torch1.8.1

激活安装的虚拟环境->进入whl的文件夹->pip install +文件名 即可完成torch的安装。

步骤三: 安装torchvision(与步骤二类似)

激活安装的虚拟环境->进入whl的文件夹->输入以下命令即可安装完成

pip install --no-deps + 文件名
1.
说明: 不添加–no-deps会导致安装时自动将你的pytorch-gpu版本自动卸载,然后装上别的版本。(这真是神奇的操作,别踩坑了。)

最后:检查是否安装成功,显示true即安装成功

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.ldbm.cn/p/439933.html

如若内容造成侵权/违法违规/事实不符,请联系编程新知网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

线性调频信号脉冲压缩

注意&#xff0c;如果用线性调频信号频谱省略前面常数因子&#xff0c;得到的脉冲压缩信号也会有一个幅度的不同

微信小程序登录与获取手机号 (Python)

文章目录 相关术语登录逻辑登录设计登录代码 相关术语 调用接口[wx.login()]获取登录凭证&#xff08;code&#xff09;。通过凭证进而换取用户登录态信息&#xff0c;包括用户在当前小程序的唯一标识&#xff08;openid&#xff09;、微信开放平台账号下的唯一标识&#xff0…

机器学习之 PCA降维

1.PCA 降维简介 主成分分析&#xff08;Principal Component Analysis, PCA&#xff09;是一种统计方法&#xff0c;用于在数据集中寻找一组线性组合的特征&#xff0c;这些特征被称为主成分。PCA 的目标是通过变换原始特征空间到新的特征空间&#xff0c;从而减少数据的维度&…

Pygame中获取鼠标按键状态的方法

在《Pygame中获取鼠标位置的方法》中提到&#xff0c;可以通过鼠标事件和mouse模块中的函数获取鼠标位置&#xff0c;这两种方法同样适用于获取鼠标按键状态。 1 通过鼠标点击事件获取鼠标按键状态 通过鼠标点击事件获取鼠标按键状态的代码如图1所示。 图1 鼠标点击事件获取鼠…

C++ | Leetcode C++题解之第390题消除游戏

题目&#xff1a; 题解&#xff1a; class Solution { public:int lastRemaining(int n) {int a1 1;int k 0, cnt n, step 1;while (cnt > 1) {if (k % 2 0) { // 正向a1 a1 step;} else { // 反向a1 (cnt % 2 0) ? a1 : a1 step;}k;cnt cnt >> 1;step …

Windows环境下VS2022编译 Speex 源码

Speex Speex 是一个开源的语音压缩格式&#xff0c;专为语音数据设计&#xff0c;提供了高压缩率的同时保持较低的比特率&#xff0c;适合网络传输。它采用了先进的编码算法&#xff0c;能够在保证声音质量的同时&#xff0c;大幅度降低文件大小&#xff0c;特别适用于实时通信…

macos安装ArgoCD

本文主要介绍如何在macos上安装并访问argo 我环境上已经安装了minikube&#xff0c;所以只需要启动minikube然后通过命令行安装argocd。 minikube start kubectl create namespace argocd kubectl apply -n argocd -f https://raw.githubusercontent.com/argoproj/argo-cd/st…

问题: java.sql.SQLException:The server time zone value ‘�й���׼ʱ��‘

原文: Mybatis PlusThe server time zone valuehis unrecognized or represents more than one time zone. You must configure either the server or JDBC driver (via the serverTimezone configuration property) to use a more specifc time zone value if you want to uti…

【全网最全】2024年数学建模国赛A题30页完整建模文档+17页成品论文+保奖matla代码+可视化图表等(后续会更新)

您的点赞收藏是我继续更新的最大动力&#xff01; 一定要点击如下的卡片那是获取资料的入口&#xff01; 【全网最全】2024年数学建模国赛A题30页完整建模文档17页成品论文保奖matla代码可视化图表等&#xff08;后续会更新&#xff09;「首先来看看目前已有的资料&#xff0…

【分支-快速排序】

【分支-快速排序】 1. 颜色分类1.1 题目来源1.2 题目描述1.3 题目解析 2. 排序数组2.1 题目来源2.2 题目描述2.3 题目解析 3. 数组中的第K个最大元素3.1 题目来源3.2 题目描述3.3 题目解析 4. 库存管理 III4.1 题目来源4.2 题目描述4 .3 题目解析 1. 颜色分类 1.1 题目来源 7…

跨越技术壁垒:EasyCVR为何选择支持FMP4格式,重塑视频汇聚平台标准

随着物联网、大数据、云计算等技术的飞速发展&#xff0c;视频监控系统已经从传统的安防监控扩展到智慧城市、智能交通、工业制造等多个领域。视频流格式作为视频数据传输与存储的基础&#xff0c;其兼容性与效率直接影响到整个视频监控系统的性能。 在众多视频流格式中&#…

UE4_后期处理_后期处理材质及后期处理体积二

效果&#xff1a; 步骤&#xff1a; 1、创建后期处理材质,并设置参数。 2、回到主界面&#xff0c;找到需要发光的物体的细节面板。 渲染自定义深度通道&#xff0c;默认自定义深度模具值为10&#xff08;需要修改此值&#xff0c;此值影响物体的亮度&#xff09;。 3、添加…

SLM561A​​系列 60V 10mA到50mA线性恒流LED驱动芯片 为智能家居照明注入新活力

SLM561A系列选型参考&#xff1a; SLM561A10ae-7G SOD123 SLM561A15ae-7G SOD123 SLM561A20ae-7G SOD123 SLM561A25ae-7G SOD123 SLM561A30ae-7G SOD123 SLM561A35ae-7G SOD123 SLM561A40ae-7G SOD123 SLM561A45ae-7G SOD123 SLM561A50ae-7G SOD123 …

java基础-IO(4)管道流 PipedOutputStream、PipedInputStream、PipedReader、PipedWriter

管道 提到管道&#xff0c;第一印象就是水管或者地下管道&#xff0c;一节接着一节&#xff0c;形如下图。 管道流 "流"从上一个管道 ——-> 下一个管道。又细分为字节流和字符流。 字节流&#xff08;PipedOutputStream、PipedInputStream&#xff09; 如果…

计算机组成原理(SRAM电路图示)

1.该电路由6个MOS管&#xff08;T1-T6&#xff09;组成 2.T1-T4是一个由MOS管组成的触发器基本电路&#xff1b; T5&#xff0c;T6像开关&#xff0c;受行地址选择信号控制&#xff1b; T7,T8受列地址选择控制&#xff0c;分别与位线A&#xff0c;和相连 3.假设触发器…

Avatar 高清图传

Avatar HD VTX 是一款数字视频发射器&#xff0c;专为与 Caddx/Walksnail 的 Avatar HD 系统配合使用而设计。最初以 Walksnail 品牌销售&#xff0c;实际上是 CaddX FPV 的一部分。 这些 VTX 设计用于 Caddx/Walksnail 的 Avatar HD 系统&#xff0c;并可与 Avatar HD Goggle…

零知识证明在BSV网络上的应用

​​发表时间&#xff1a;2023年6月15日 2024年7月19日&#xff0c;BSV区块链主网上成功通过使用零知识证明验证了一笔交易。 零知识证明是一种技术&#xff0c;它允许一方&#xff08;证明者&#xff09;在不透露任何秘密的情况下&#xff0c;向另一方&#xff08;验证者&…

ThinkPHP5 5.0.23-rce远程代码执行漏洞复现

启动环境&#xff0c;先关闭其他环境 启动 判断是否存在漏洞&#xff1a;访问/index.php?scaptcha页面&#xff0c;会出现报错 使用HackBar 插件发送 POST 请求 _method__construct&filter[]system&methodget&server[REQUEST_METHOD]dir 通过echo命令写入 Webshe…

2024Java基础总结+【Java数据结构】(2)

面向对象07&#xff1a;简单小结类与对象 面向对象08&#xff1a;封装详解 面向对象09&#xff1a;什么是继承 ctrlh看类的关系&#xff0c;所有的类都默认的或间接继承Object 面向对象10&#xff1a;Super详解 super注意点: super调用父类的构造方法&#xff0c;必须在构造方…

JavaScript使用高德API显示地图

前言 在JavaScript中&#xff0c;使用Leaflet库显示地图是一种常见的做法。Leaflet是一个开源的JavaScript库&#xff0c;用于在Web应用程序中创建互动地图。它非常轻量级&#xff0c;易于使用&#xff0c;并且提供了多种功能&#xff0c;使开发者能够轻松地将地图集成到他们的…