GPU版本torch使用教程

GPU版本torch使用教程

一、下载配置CUDA和CUDNN

(1)进入cmd使用nvidia-smi.exe查看自己电脑支持的最新CUDA版本(可以下载地低版本),如图:
在这里插入图片描述
也可以通过NVIDIA控制面板(NVIDIA Control Panel)查看。
在这里插入图片描述
(2)下载所需要的cuda可执行文件
CUDA指定版本下载网址:https://developer.nvidia.com/cuda-toolkit-archive
请根据需要选择对应版本
安装CUDA
1.运行可执行文件,注意!
在这里插入图片描述
上面选择的并不是安装位置!而是临时的文件存储位置,安装完成会自动删除。
1.1自选安装位置,需要选择自定义,如图:
在这里插入图片描述
然后选择自己想要安装的路径即可(两个可以选在同一个文件夹中)
在这里插入图片描述
1.2无脑下一步即可,默认安装位置为

C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.1

(3)下载CUDnn文件包
下载网址:https://developer.nvidia.com/rdp/cudnn-archive
解压文件包,将以下三个文件复制到CUDA安装位置即可
在这里插入图片描述

二、下载pytorch安装需要的.whl文件

下载网址:https://download.pytorch.org/whl/torch_stable.html
torch对应的.whl文件下载
在这里插入图片描述
torchvision 同理
在这里插入图片描述

三、安装torch和torchvision

使用命令

pip install ".whl文件路径"

四、测试

测试语句:

import torch
torch.cuda.is_available()

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

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

相关文章

JAVA MQTT 发布主题请求,订阅主题接收,订阅主题回复,发布主题再接收回复,三步走

先看效果 一、准备工作 1.官网下载emqx压缩包放到自己的盘符下,不要带中文路径 下载 EMQX 2.在路径的bin中,cmd,启动emqx服务 emqx start 3.访问服务,能打开就证明启动成功,登录的话官网默认的密码账号(…

实现SpringMVC底层机制(二)

文章目录 1. 动态获取spring配置文件1.修改SunWebApplicationContext.java2.修改SunDispatcherServlet.java 2.自定义Service注解1.需求分析2.编写Monster.java3.自定义Service注解4.编写Service接口MonsterService.java5.编写Service实现类MonsterServiceImpl.java6.修改SunWe…

linux系统下载huggingface大模型教程

文章目录 准备工作添加SSH Key生成Access Token 模型下载公开模型下载(bert-base-chinese为例)非公开模型下载(Llama3为例)权限申请官网预训练模型下载huggingface仓库下载 准备工作 添加SSH Key # 本地机器生成ssh key # step1…

JavaScript-3(内置对象+数组对象+字符串对象)

目录 1.预解析 2.对象 什么是对象 创建对象的三种方法 利用字面量创建方法 利用new Object创建对象 构造函数创建对象 new关键字 遍历对象 3.内置对象 Math对象 Math概述 Math随机数 Date日期对象 格式化日期 Date总的时间毫秒 4.数组对象 创建数组的两种方式…

OpenHarmony实战开发-按钮 (Button)

Button是按钮组件,通常用于响应用户的点击操作,其类型包括胶囊按钮、圆形按钮、普通按钮。Button做为容器使用时可以通过添加子组件实现包含文字、图片等元素的按钮。具体用法请参考Button。 创建按钮 Button通过调用接口来创建,接口调用有…

JetBrains CLion v2023.3.4 激活版 (C/C++ 集成开发IDE)

前言 JetBrains CLion是一款跨平台的C/C集成开发环境,由JetBrains公司推出。其最新版本支持C14几乎完全,并初步支持C17,使得编写代码更加便捷。CLion还提供了Disassembly view(反汇编视图),即使没有源代码…

C++对象的初始化和处理

生活中我们买的电子产品都基本会有出厂设置!在某一天我们不用时候也会删除一些自己信息数据保证安全。 C中的面向对象来源于生活,每个对象也都会有初始设置以及对象销毁前的清理数据的设置。 构造函数和析构函数 对象的初始化和清理也是两个非常重要的安全问题 一…

CSS中设置透明度的2个属性:opacity,RGBA以及它们的区别

你好,我是云桃桃。 一个希望帮助更多朋友快速入门 WEB 前端的程序媛。 云桃桃-大专生,一枚程序媛,感谢关注。回复 “前端基础题”,可免费获得前端基础 100 题汇总,回复 “前端工具”,可获取 Web 开发工具合…

什么是停车场车牌识别系统,停车场车牌识别系统工作原理是什么

在现代城市中,随着机动车辆数量的急剧增加,停车场道闸系统的智能化管理变得尤为重要。传统的人工管理方式不仅效率低下,而且容易出错,无法满足日益增长的车辆管理需求。车牌识别技术的引入,为停车场道闸系统带来了革命…

python 编程小技巧:# type: 类型注释语法

# type: 是 Python 3.5 引入的一种类型注释语法,用于在代码中指定变量、函数、方法等对象的类型信息,以便 IDE 和类型检查工具等工具能够更好地理解和分析代码。具体来说,# type: 后面可以跟一个类型注释,用于指定对象的类型&…

小程序变更主体还要重新备案吗?

小程序迁移变更主体有什么作用?小程序迁移变更主体的作用可不止变更主体这一个哦!还可以解决一些历史遗留问题,比如小程序申请时主体不准确,或者主体发生合并、分立或业务调整等情况。这样一来,账号在认证或年审时就不…

42-巩固练习(二)

42-1 函数的递归 1、问:关于递归的描述错误的是() A.存在限制条件,当满足这个限制条件的时候。递归便不再维续 B.每次递归调用之后越来越接近这个限制条件 C.递归可以无限递归下去 D.递归层次太深,会出现栈溢出现…