[Howto] Pytorch Window GPU 环境配置

news2025/7/28 5:59:45

注: how to 系列只有基本的操作,不涉及原理,类似于操作手册。

Linux,MacOS和Window配置方法差不多,就是环境变量的修改方式不太一样,就不单独说明了。

1. 确定Pytorch版本

估计像我一样的初学者,很多时候都是参考其他人的代码,再进行扩展开发,所以pytorch的版本一开始和参考对象是一致的。可以从下面的链接中寻找自己需要的版本。

https://pytorch.org/get-started/locally/

页面中也有previous 版本,也可以进行查看

在这里插入图片描述
本文以 pytorch 1.13.0为例,选择pytroch版本。然后就可以看到需要CUDA支持什么版本了,选择完之后,就可以看到命令,进行一键安装了。我这边介绍详细的安装步骤。

2. 安装显卡驱动

2.1 确定需要驱动版本

已CUDA 11.7 为例,先从官网查看显卡驱动的最低版本

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

在这里插入图片描述
在这里插入图片描述

如图所示,window下,CUDA11.7,显卡驱动大于516.31就行。

2.2 检查驱动

输入nvidia-smi 查看显卡型号,和目前的显卡驱动和能支持的CUDA最高版本。

在这里插入图片描述
如上图,我显卡型号是GeForce RTX 2070,目前的显卡驱动版本是431.6,那么需要安装更新版本的显卡驱动,如果当前的显卡驱动版本大于516.31,则不需要安装新版本的显卡驱动。

2.3 更新显卡驱动

可以去NVIDIA官网根据自己的显卡型号,进行驱动的更新,

NVIDIA® GPU drivers
https://www.nvidia.com/download/driverResults.aspx/193884/en-us/

按照自己的实际情况填表

在这里插入图片描述
从结果列表里面,选择一个版本就行了,我选择了522.30版本,是大于之前要求的516.31版本的。

在这里插入图片描述

3 安装CUDA

3.1 确定需要版本

由之前的pytorch的选择图中,可以看到需要11.7版本

3.2 检查当前版本

输入nvcc --version,可以查看目前的cuda版本,如下图所示,我这边是10.0版本,不符合之前的11.7的需求

在这里插入图片描述
版本低于所需版本,所以要更新CUDA版本

3.3 更新CUDA

官网下载即可

https://developer.nvidia.com/cuda-toolkit-archive在这里插入图片描述

https://developer.nvidia.com/cuda-11-7-1-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

按需选择,
在这里插入图片描述

4. cuDNN

https://developer.nvidia.com/cudnn

需要加入程序员计划
https://developer.nvidia.com/rdp/cudnn-download

在这里插入图片描述

拷贝到CUDA的目录就行。

5. 安装pytorch

都安装完成,检查:
在这里插入图片描述

然后使用最开始的一句话的安装命令就行,会发现安装得非常快。

import torch
print(torch.cuda.is_available())

运行,检查,显示true,则安装成功。

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

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

相关文章

关于commonjs、AMD、UMD、ESM以及ts模块之间的使用差异

commonjs 特点:一个文件就是一个模块,拥有独立的作用域,适用于服务端不适合浏览器端。导出模块内部数据通过module.exports或exports对象默认导出: // true const a 1 const b 2 module.exports {a, b }或者 // true const …

【跨境电商卖家】Instagram营销初学者指南(二):方法与技巧

关键词:跨境电商卖家、instagram营销 1.为 Instagram营销设定目标 在你开始在 Instagram 上发帖之前,问问你自己(或你的团队)一件事:你为什么在 Instagram 上?尽管该平台很受欢迎,但您的回答不…

linux笔记(6):东山哪吒D1H显示HDMI测试-命令行调试

文章目录1.测试流程和结果2.测试过程详解2.1 挂载测试工具1.2 设置参数1.2.1设置name1.2.2选择命令1.2.3 设置命令参数1.3开启显示3.还没搞清楚怎么在应用中显示字符测试开发板的HDMI输出。 参考文档:全志官方文档。 1.测试流程和结果 测试结果: 2.测…

如何实现一键全选

利用复选框的激活、非激活实现一键全选功能 效果展示 前置准备 投票列表素材 具体步骤 添加素材 制作列表复选框 制作一件全选按钮 创建复选框相关行为触发器 制作一键全选触发器 步骤分解 添加素材 拖拽 图片组件 到 根容器 选中 图片组件 铺满父级容器 点击 检查面板 中的 …

一种新的群体智能优化算法:麻雀搜索算法(SSA)(Matlab代码实现)

🍒🍒🍒欢迎关注🌈🌈🌈 📝个人主页:我爱Matlab 👍点赞➕评论➕收藏 养成习惯(一键三连)🌻🌻🌻 🍌希…

C++中函数调用的整个过程内存堆栈分配详解

函数调用过程中:实参将值拷贝给函数的形参,而函数的形参相当于一个生存周期位于函数 内部的局部变量,函数内部的内存操作也只是将拷贝到形参的值进行操作,形参在函数结束 后会被栈自动回收释放(形参在栈中分配),这就是…

Spring Security如何防止会话固定攻击

在春季安全课程的这篇文章中,我们将研究春季安全会话固定以及如何防止春季应用程序中的会话劫持。 春季安全会话固定 会话固定是一种非常常见且最常见的攻击类型,恶意攻击者可以通过访问站点来创建会话,然后诱使其他用户使用相同的会话登录…

副业是刚需?分享几个程序员接外包私活的网站

经常看到某某程序员接了个项目开发,工作之余轻轻松松赚了钱还顺带提升了技术;或者看到某大佬又发表了一篇程序员技术提升稿件,阅读点赞收藏三连发,这个月的零花钱又不愁了...但自己只是一名普普通通的程序员,能找到这样…

Golang入门笔记(10)—— 闭包 closure

先看一段代码,脱离代码讲闭包,太干了。 package mainimport "fmt"func main() {a : Adder()fmt.Println(a(1))fmt.Println(a(2))fmt.Println(a(3)) }func Adder() func(int) int { // 累加器:这里从10开始累加var sum int 10retu…

linux时区相关

背景:用linux自带的时间接口函数读取时间的时候,发现有时候时间与北京时间不符合,经过研究发现:时间 UTC时间时区带来的偏移。操作方法:timedatectl list-timezones可看支持的时区改时区方法有如下两种: l…

【LeetCode 每日一题】15. 三数之和

01 题目描述 给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i ! j、i ! k 且 j ! k ,同时还满足 nums[i] nums[j] nums[k] 0 。请 你返回所有和为 0 且不重复的三元组。 注意:答案中不可以包含重复的三元…

【数据结构】链表LinkedList

1.ArrayList的缺陷 2.单链表的实现 3.LinkedList的使用(模拟实现) 我们之前介绍过ArrayList了,它的底层是数组,数组是一段连续的空间,当我们想要插入或者删除数据的时候,插入元素,就要让插入位置…

用树莓派PICO做一个桌面时钟超详细教程!

用树莓派PICO做一个可显示时间和温湿度的桌面时钟一、概述二、材料准备1、树莓派PICO2、DHT11温湿度传感器3、DS1302时钟模块(选用)4、SSD1306屏幕5、其他材料三、开始1、连线2、写程序(1)使用内置RTC函数实现的时钟(2…

2.11 教你一套怎么建立自己的选题素材库的方法【玩赚小红书】

一、自身定位延伸选题库 建立选题库,前提先确定自身定位,然后发散性思维延展。如我们做母婴博主,接下来就要想,母婴用户会关注什么,自然会想到到宝宝吃喝玩乐、妈妈保养、产后修复、婆媳关系等等内容。若我们只做宝宝…

这才是,真彩虹预染蛋白Markers

做WB的小伙伴都知道,现市面上各种“多彩”Marker的产品有很多,但是真正拿到手上的,可能是各种各样的(见图1),咱也不清楚哪个是真的... 现在小编告诉你,经典的彩虹Marker长这样(见图2)&#xff1…

WebDAV之葫芦儿·派盘+读出通知

读出通知 支持webdav方式连接葫芦儿派盘。 手机各种推销通知太多,如何避免那些繁琐的通知内容,做出一键就能够阅读重要通知的最佳体验,帮助您更加快速和便捷的体验到那些应用内容?推荐大家使用读出通知。 读出通知APP可以设置接收通知的app,还可以用耳机操作,操作简单…

avalanche 少量tcp长连接持续构建HTTP请求测试

最近测试项目,测试要求使用少量tcp长连接连接,持续打HTTP请求,到测试结束。 分别用思博伦测试仪和supernova测试仪进行实现。 思博伦测试仪实现 测试仪基本运行流程:Loads配置任何形式bandwidth,connection&#xf…

SpringBoot项目本机和Linux环境部署

文章目录一. 本机环境下打包与运行二. Linux下部署SpringBoot项目2.1 Linux环境配置2.2 配置数据库2.3 运行程序一. 本机环境下打包与运行 项目进行打包 2. 本机环境下运行SpringBoot程序 控制台进入SpringBoot项目jar包所在的文件夹,运行下面指令即可 java -jar […

[附源码]java毕业设计企业招标系统

项目运行 环境配置: Jdk1.8 Tomcat7.0 Mysql HBuilderX(Webstorm也行) Eclispe(IntelliJ IDEA,Eclispe,MyEclispe,Sts都支持)。 项目技术: SSM mybatis Maven Vue 等等组成,B/S模式 M…

JDBC技术

JDBC 一、jdbc的概述 JDBC(Java Database Connectivity)是一个独立于特定数据库管理系统、通用的SQL数据库存取和操作的公共接口(一组API),定义了用来访问数据库的标准Java类库,(java.sql,javax.sql)使用…