PyTorch保姆级安装教程

news2024/9/8 23:17:07

1 安装CUDA

1.1 查找Nvidia适用的CUDA版本

桌面右键,【打开 NVIDIA控制面板】

查看【系统信息】

查看NVIDIA的支持的CUDA的版本,下图可知支持的版本是 10.1

1.2 下载CUDA

CUDA下载官方网址

https://developer.nvidia.com/cuda-toolkit-archive

找到适合的版本下载

选择适合自己的下载,这里选择 exe(local)下载,我这里不小心标记错误了

1.3 安装CUDA

下载完毕双击.exe安装包安装即可

安装的时候留意CUDA的安装路径,下面要用得到。

我安装的时候,安装完毕之后环境变量是自动配置好的,如下图:

如果你的没有的话,加上这四个环境变量即可。一般来说是自动配置好的。

2 安装cuDNN

2.1 下载cuDNN

官网下载cuDNN地址

找到适合自己的版本

点击之后,选择windows版本的

直接下载可能下载比较慢,这里可以通过百度网盘下载,链接在下

链接:https://pan.baidu.com/s/16UKQMX3ww-9GCFn3dYG5CA

提取码:3535

2 安装cuDNN

下载完毕之后,解压安装包

复制上面的三个文件到CUDA的安装路径,直接覆盖原文件即可。如果你的版本和我的不一样,只需要找到你自己的版本,步骤没有变化。

3 安装PyTorch

3.1 创建PyTorch虚拟环境

如果你还没有安装Anaconda,需要先去安装一下,可以参照

windows下的Anaconda的安装和配置 进行安装anaconda。

anaconda安装完毕之后,打开Anaconda Prompt

创建一个Python的虚拟环境来安装Pytorch

conda create -n PyTorch python=3.9

其中【Pytorch】是虚拟环境的名称,可以和我的不一样,但是建议不要太繁琐。

创建完毕之后,会提示你如何激活虚拟环境

虚拟环境的创建先到这里。

4 下载torch和torchvision

下载地址

打开下载地址是这样的

我们往下翻,找到cuda版本的

我们根据cuda的版本和python的版本选择下载,这里我选择的是torch-1.7.1版本。

接着我们下载torchvision,这里是torch和torchvision的版本适配图

因为上面我选择的是torch1.7.1版本,所以这里我的torchvision就要下载0.8.2版本的

把这两个文件下载之后放到指定文件即可。稍后安装。

5 终章

进入到之前创建的虚拟环境PyTorch并激活,然后在虚拟环境中,切换到你的torch和torchvision的目录下,切换目录要一层一层切换

然后pip install刚才的两个文件即可

安装成功之后,测试一下

打印出 True 证明安装成功。

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

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

第六章 effect.scheduler功能实现

effect.scheduler功能实现 主要先了解scheduler需要实现什么样的需求,有一下四点: 1 通过 effect 的第二个参数给定一个 scheduler 的 fn 2 effect 第一次执行的时候 还会执行 fn 3 当 响应式对象 set update 不执行fn 而是执行 scheduler 4 如果说…

面试问题【线程】

线程什么是进程什么是线程进程和线程的关系什么是并发和并行如何使用线程Thread 和 Runnable 两种开发线程的区别线程的生命周期什么是上下文切换什么是线程死锁如何避免死锁说说 sleep() 方法和 wait() 方法区别和共同点为什么我们调用 start() 方法时会执行 run() 方法&#…

Transformer学习

原论文:Attention Is All You Need。论文地址:https://arxiv.org/abs/1706.03762. Transformer是针对自然语言处理的,Google在2017年发表在Computation and Language,RNN模型记忆长度有限且无法并行化但是Tranformer解决了上述问…

解析几何北大第五版复习提纲

第一章 两向量向量积 向量积定义:a x b |a||b|sin几何意义:平行四边形面积性质: 两向量共线的充分必要条件是 a x b 0 数乘: 分配律: 求法:行列式 三向量混合积 混合积定义:对于一个六面体,…

快鲸SCRM发布口腔企业私域运营解决方案

口腔企业普遍面临着以下几方面运营痛点问题 1、获客成本居高不下,恶性竞争严重 2、管理系统落后,人员流失严重 3、客户顾虑多、决策时间长 4、老客户易流失,粘性差 以上这些痛点,不得不倒逼口腔企业向精细化运营客户迈进。 …

【LeetCode】剑指 Offer 21. 调整数组顺序使奇数位于偶数前面 p131 -- Java Version

题目链接:https://leetcode.cn/problems/diao-zheng-shu-zu-shun-xu-shi-qi-shu-wei-yu-ou-shu-qian-mian-lcof/ 1. 题目介绍(21. 调整数组顺序使奇数位于偶数前面) 输入一个整数数组,实现一个函数来调整该数组中数字的顺序&…

4自由度串联机械臂按颜色分拣物品功能的实现

1. 功能说明 本实验要实现的功能是:将黑、白两种颜色的工件分别放置在传感器上时,机械臂会根据检测到的颜色,将工件搬运至写有相应颜色字样区域。 2. 使用样机 本实验使用的样机为4自由度串联机械臂。 3. 运动功能实现 3.1 电子硬件 在这个…

快速吃透π型滤波电路-LC-RC滤波器

π型滤波器简介 π型滤波器包括两个电容器和一个电感器,它的输入和输出都呈低阻抗。π型滤波有RC和LC两种, 在输出电流不大的情况下用RC,R的取值不能太大,一般几个至几十欧姆,其优点是成本低。其缺点是电阻要消耗一些…

亚马逊二审来袭,跨境电商传统验证算法真的靠谱吗?

多个大卖突遭二审 已有卖家账号被封 近期有不少卖家在论坛上反映称自己收到了亚马逊的二次视频验证邮件。 邮件上称: 卖家必须要完成额外的身份审查,才有资格在亚马逊继续销售商品;亚马逊要求卖家出示注册时提交的身份证原件和营业执照原件…

聊聊混沌工程

这是鼎叔的第五十四篇原创文章。行业大牛和刚毕业的小白,都可以进来聊聊。欢迎关注本专栏和微信公众号《敏捷测试转型》,大量原创思考文章陆续推出。混沌工程是一门新兴学科,它不仅仅只是个技术活动,还包含如何设计能够持续协作的…

xgboost:算法数学原理

xgboost算法数学原理 1、求预测值 y^iϕ(xi)∑k1Kfk(xi),fk∈F,(1)\hat{y}_i\phi\left(\mathbf{x}_i\right)\sum_{k1}^K f_k\left(\mathbf{x}_i\right), \quad f_k \in \mathcal{F},\tag{1} y^​i​ϕ(xi​)k1∑K​fk​(xi​),fk​∈F,(1) F{f(x)wq(x)}(q:Rm→T,w∈RT)\mathca…

即时通讯开发常用加解密算法与通讯安全

平时开发工作中,我们会经常接触加密、解密的技术。尤其在今天移动互联网时代,越来越多的用户会将数据存储在云端,或使用在线的服务处理信息。这些数据有些涉及用户的隐私,有些涉及用户的财产,要是没有一套的方案来解决…

玩转结构体---【C语言】

⛩️博主主页:威化小餅干📝系列专栏:【C语言】藏宝图🎏 ✨绳锯⽊断,⽔滴⽯穿!一个编程爱好者的学习记录!✨目录结构体类型的声明结构体成员访问结构体传参前言我们是否有想过,为什么会有结构体呢…

stylelint执行插件的全过程

stylelint可以用来扩展插件去实现各种规则,接下来带大家看看stylelint是如何执行插件的 首先遍历absoluteFilePaths路径(该路径是我们执行lint命令配置的文件类型eslint --cache --max-warnings 0 \"{src,mock}/**/*.{vue,ts,tsx}\" --fix&…

Spark RDD

RDD RDD 是构建 Spark 分布式内存计算引擎的基石,如 :DAG/调度系统都衍生自 RDD RDD 是对分布式数据集的抽象,囊括所有内存/磁盘的分布式数据实体 RDD/数组差异 对比项数组RDD概念数据结构实体数据模型抽象数据跨度单机进程内跨进程&…

GC简介和监控调优

GC简介: GC(Garbage Collection)是java中的垃圾回收机制,是Java与C/C的主要区别之一,在使用JAVA的时候,一般不需要专门编写内存回收和垃圾清理代 码。这是因为在Java虚拟机中,存在自动内存管理和垃圾清扫机制。 什么…

ThreadLoca基本使用以及与synchronized的区别

文章目录1. ThreadLocal介绍1.1 官方介绍1.2 基本使用1.2.1 常用方法1.2.2 使用案例1.3 ThreadLocal类与synchronized关键字1.3.1 synchronized同步方式1.3.2 ThreadLocal与synchronized的区别2. 运用场景_事务案例2.1 转账案例2.1.1 场景构建2.1.2 引入事务2.2 常规解决方案2.…

k8s servelList(服务列表) 卡死不同步问题分析

提要容器集群版本情况:k8s 1.20客户端k8s client版本: 0.21事情是这样的,运行了一年的服务,突然有一天业务反馈服务使用异常,然后初步调查结果如下以下截图是网关异常以下截图是客户端zull(feign&#xff0…

依赖倒置DIP在系统架构中的应用

最近在对项目中的某一模块进行重构和功能的拓展。一直没想到好方法。 简单理解为: R项目 调用了 E项目的打印接口,但是E项目需要对R传来对数据传输对象DTO进行二次处理,甚至夹杂很多R项目的业务逻辑(去调用R项目的接口&#xff0…

代码规范书写说明

目录 一,命名风格 二、常量定义 三、代码格式 一,命名风格 (1)、不能够以下划线或者美元符号开始,也不能以下划线或者美元符号结束 反例:_name / __name / $name / name_ (2)、所…