windows10系统安装cuda版pytorch的一般步骤

news/2025/3/25 5:08:57/文章来源:https://www.cnblogs.com/rainbow70626/p/18787902

windows10系统安装cuda版pytorch的一般步骤:
1、首先,查看本机是否有nvidia的GPU卡,并确定nvidia的GPU卡的型号。
2、其次,去nvidia官网下载对应显卡的最新版的驱动,下载地址为:https://www.nvidia.cn/drivers/lookup/。
下载驱动之后完成安装,之后在操作系统桌面右键单击-nvidia控制面板-系统信息-组件下面的NVCUDA64.dll一行可以看到该版本驱动支持的CUDA的最高版本,如我的电脑显示的是CUDA 11.4.557。
3、接着,去nvidia官网下载对应版本的CUDA Toolkit并行计算工具包,如适配本机显卡的CUDA Toolkit 11.4.4,下载地址为:https://developer.nvidia.com/cuda-toolkit-archive。
下载之后完成安装。并配置好CUDA的环境变量。配置方法参考链接:https://www.cnblogs.com/rainbow70626/p/6498714.html。
4、再次,去nvidia官网下载与CUDA Toolkit并行计算工具包对应的cuDNN库,Download cuDNN v8.2.4 (September 2nd, 2021), for CUDA 11.4,下载地址为:https://developer.nvidia.com/rdp/cudnn-archive。
下载之后完成安装。安装方法参考链接:https://blog.csdn.net/weixin_44779079/article/details/141528972。
5、下载cuda版pytorch离线安装包,下载地址为:https://download.pytorch.org/whl/torch/或https://download.pytorch.org/whl/torch_stable.html。比如我下载的是torch-1.12.1+cu113-cp39-cp39-win_amd64.whl,该文件的文件名中torch-1.12.1代表1.21.1版的torch,cu113代表cuda11.3版本,cp39-cp39代表python3.9版本,
6、下载Anaconda,下载地址为:Anaconda官网https://www.anaconda.com/download或清华大学镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/,本机下载的是Anaconda3-2022.10-Windows-x86_64.exe。为啥下载该版本的Anaconda3?是因为它内嵌的python版本是3.9,与前一步下载的cuda版pytorch安装包项匹配。
这一过程可以在安装Anaconda3过程中,Anaconda安装包界面上看到python版本。下载Anaconda之后完成安装。
7、打开Anaconda prompt创建一个虚拟环境、激活虚拟环境、安装离线版的pytorch、验证gpu版pytorch是否安装成功。

#pytorch各个版本下载地址
pytorch的官方下载地址:https://download.pytorch.org/whl/
#anaconda各个版本下载地址
清华大学开源软件镜像站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
一、离线安装
0、从链接https://download.pytorch.org/whl/torch/或https://download.pytorch.org/whl/torch_stable.html下载对应的CUDA版本的pytorch并保存到指定的文件夹。
1、打开Anaconda prompt创建一个虚拟环境:名为pytorch
conda create --name pytorch python=3.6 anaconda
2、激活该虚拟环境
activate pytorch
3、安装torch。找到之前下载的pytorch的绝对路径,E:\torch-1.7.0+cu101-cp36-cp36m-win_amd64.whl,则直接用以下命令安装:
pip install E:\torch-1.7.0+cu101-cp36-cp36m-win_amd64.whl
4、上述安装过程中可能需要联网下载某些python包,直至提示pytorch安装成功。
5、判断gpu版pytorch是否安装成功。首先输入命令python切换到python运行环境,然后在输入以下程序,根据程序运行结果判断gpu版pytorch。
import torch
torch.cuda.is_available()
print("PyTorch version:",torch.__version__)

二、在线安装
去pytorch官网按命令行下载,不过比较慢,不推荐,传送门
1、根据自己的电脑(windows,Linux,mac选择)
2、使用conda安装
3、python编译环境
4、cuda版本号,这个根据自己之前安装的cuda版本号来确定
由于我的cuda:11.3,所以就选择这个,
下载:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
原文链接:https://blog.csdn.net/HYilalala/article/details/121128944
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
# 安装时指定版本号
# CUDA 11.3
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
ERROR: Could not find a version that satisfies the requirement torch==xxx+cuxxx解决方案:
1、https://blog.csdn.net/weixin_46653651/article/details/140936271
2、https://blog.csdn.net/leva345/article/details/119836011

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

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

相关文章

Pydantic字段级校验:解锁@validator的12种应用

title: Pydantic字段级校验:解锁@validator的12种应用 date: 2025/3/23 updated: 2025/3/23 author: cmdragon excerpt: Pydantic校验系统支持通过pre验证器实现原始数据预处理,在类型转换前完成字符清洗等操作。格式验证涵盖正则表达式匹配与枚举值约束,确保护照编号等字…

20244126 2024-2025-2 《python程序设计》实验一报告

课程:《Python程序设计》 班级:2441 姓名:马晓霞 学号:20244126 实验教师:王志强 实验日期:2025年3月23日 必修/选修:公选课 (一)实验内容 1.熟悉Python开发环境 2.练习Python运行,调试技能 3.编写技能,练习变量和类型、字符串、对象、缩进和注释等 4.编写一个猜数字…

1.6K star!这个开源文本提取神器,5分钟搞定PDF/图片/Office文档!

Kreuzberg 是一个基于 Python 的文本提取库,支持从 PDF、图像、Office 文档等 20+ 格式中提取文本内容。采用 MIT 开源协议,具备本地处理、异步架构、智能 OCR 等特性,特别适合需要隐私保护的文档处理场景。嗨,大家好,我是小华同学,关注我们获得“最新、最全、最优质”开…

leetcode每日一题:判断一个括号字符串是否有效

题目 一个括号字符串是只由 ( 和 ) 组成的 非空 字符串。如果一个字符串满足下面 任意 一个条件,那么它就是有效的:字符串为 (). 它可以表示为 AB(A 与 B 连接),其中A 和 B 都是有效括号字符串。 它可以表示为 (A) ,其中 A 是一个有效括号字符串。给你一个括号字符串 s 和…

实验2c语言分支与循环基础应用编程

任务一:1 #include <stdio.h>2 #include <stdlib.h>3 #include <time.h>4 5 #define N 56 7 int main() {8 int number;9 int i; 10 11 srand(time(0)); // 以当前系统时间作为随机种子 12 for(i = 0; i < N; ++i) { 13 nu…

矿用人员违规闯入监控报警系统

矿用人员违规闯入监控报警系统,设置警戒预警功能,巷道正在行车时,当有行人闯入时,及时抓拍、识别现场违章入侵人员同时闪光警示和音箱报警提示禁止进入,在绞车运行期间人员误入斜巷能够及时报警,自动停止绞车运行。该系统做到绞车运行和行人的安全隔离,有力保障了煤矿斜…

工厂车间人员违规闯入禁区报警系统

工厂车间人员违规闯入禁区报警系统前端设备嵌入AI人体识别算法,对人员的检测、跟踪,实现对人体检测分析识别,实时预警周界区域内人员入侵事件。当有可疑人员进入监测范围内可对其自动识别,即对其抓拍并将当时图像传输到管理中心,在管理中心输出报警信号。工厂车间人员违规…

iis备份还原工具,想要备份iis有哪些工具可以实现?

想要备份IIS(Internet Information Services),可以使用多种专门的备份还原工具。以下是一些常用的IIS备份还原工具及其简要介绍:IISBackUp 功能:专注于备份和恢复IIS站点设置,包括网站配置文件、站点绑定信息、应用程序池设置、SSL证书和密钥信息等。 特点: 提供一种方便…

集合的通用遍历方法--java进阶day09

1.集合的三种通用遍历方法之前我们学习过集合的遍历方法,为什么这里还要再学呢? 这是因为,之前我们用的遍历方法使用了索引,但我们知道set接口的实现类的集合均无索引,所以我们要学习通用的遍历方法 2.迭代器遍历 1.迭代器 迭代器叫做Iterator,是一个接口,我们知道接口是…

安卓逆向:adb连接模拟器进行jeb动态调试

这篇文章的内容记录下我在各个文章搜索学习最后解决了安卓进行模拟器动态调试的方法。 参考的文章有: ADB:https://blog.csdn.net/Python_0011/article/details/132040387 ADB的使用https://blog.csdn.net/SM1555/article/details/112608576 Androidkiller:https://blog.csdn.…

Java中的Map CAS AQS

Java中的Map 1.基本介绍和api使用就免了 Java中的Map是一种用于存储键值对(Key-Value)的接口,属于java.util包,是集合框架的重要组成部分。 2.HashMap从图中的关系可以看出这些类间关系了。 ①基本分析 HashMap的一些属性 // 默认容量 16 static final int DEFAULT_INITIAL…

一款 .NET 开源、功能强大的远程连接管理工具,支持 RDP、VNC、SSH 等多种主流协议!

前言 今天大姚给大家分享一款基于 .NET 开源(GPL-2.0 license)、免费、功能强大的 Windows 远程连接管理工具,支持 RDP、VNC、SSH 等多种主流协议:mRemoteNG。 项目介绍 mRemoteNG是一款基于 .NET 开源(GPL-2.0 license)、免费、功能强大的 Windows 远程连接管理工具,支…