解决RuntimeError: CUDA error: invalid device ordinal

步骤

首先查看自己设备的cuda版本

#如下linux指令都可以,主要还是以nvidia-smi为主
nvidia-smi
nvcc -V

看了其他博文是可以向下兼容
用的python版本是3.8
torch版本用的1.12.1+cu113
torch网址:https://pytorch.org/get-started/previous-versions/

在这里插入图片描述

安装完后发现出现如下问题:
在这里插入图片描述

import torch
print(torch.__version__)                # 查看pytorch安装的版本号
print(torch.cuda.is_available())        # 查看cuda是否可用。True为可用,即是gpu版本pytorch
print(torch.cuda.get_device_name())    # 返回GPU型号
print(torch.cuda.device_count())        # 返回可以用的cuda(GPU)数量,0代表一个
print(torch.version.cuda)   

运行发现GPU数量为1,原本应该是8.

解决方案

在linux指令里面输入:

#GPU的数量export CUDA_VISIBLE_DEVICES=[GPU的个数]
#eg. 8卡就用下语句
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
#4卡
export CUDA_VISIBLE_DEVICES=0,1,2,3

重新运行后可以看到GPU的数量为8.
在这里插入图片描述

export CUDA_VISIBLE_DEVICES=device_ids例如,要将环境变量设置为使用设备 01,可以执行以下命令:export CUDA_VISIBLE_DEVICES=0,1如果要使用所有可见的设备,可以将 `device_ids` 设置为空:export CUDA_VISIBLE_DEVICES=这将清除 `CUDA_VISIBLE_DEVICES` 环境变量,使所有可见的设备都可用。

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

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

相关文章

前端路由钩子的神奇之处:你真的了解它们吗?(上)

🤍 前端开发工程师(主业)、技术博主(副业)、已过CET6 🍨 阿珊和她的猫_CSDN个人主页 🕠 牛客高级专题作者、在牛客打造高质量专栏《前端面试必备》 🍚 蓝桥云课签约作者、已在蓝桥云…

MySQL按序批量操作大量数据

MySQL按序批量操作大量数据(Java、springboot、mybatisplus、ElasticSearch) 以同步全量MySQL数据到ElasticSearch为例。 核心代码 业务逻辑: public boolean syncToElasticsearch() {log.info("Starting data synchronization to El…

第13节:Vue3 声明反应状态ref()

下面是一个示例&#xff0c;演示了如何在UniApp中使用Vue3框架使用声明反应状态ref()&#xff1a; vue复制代码<template> <view> <text>{{ message }}</text> <button click"changeMessage">点击改变文本</button> </v…

CSS 基础

文章目录 CSS 常见的属性CSS 常见样式行内样式内嵌样式导入样式 CSS 选择器标签选择器id选择器类选择器全局选择器属性选择器组合选择器 CSS 常见应用表格列表导航栏下拉菜单提示工具图片廊 CSS (Cascading Style Sheets&#xff0c;层叠样式表&#xff09;&#xff0c;是一种用…

数据库连接池Druid

在 Spring Boot 项目中&#xff0c;数据库连接池已经成为标配&#xff0c;然而&#xff0c;我曾经遇到过不少连接池异常导致业务错误的事故。很多经验丰富的工程师也可能不小心在这方面出现问题。 在这篇文章中&#xff0c;我们将探讨数据库连接池&#xff0c;深入解析其实现机…

Arthas 排查JAVA应用问题 —— 筑梦之路

简介 Arthas 是阿里开源的一款 Java 应用诊断工具&#xff0c;可以在线排查问题&#xff0c;动态跟踪 Java 代码&#xff0c;以及实时监控 JVM 状态。 官方网站&#xff1a;arthas 快速入门 1. 下载 wget https://arthas.aliyun.com/arthas-boot.jar 2. 运行 需要在运行…

【细说Java内部类】

系列文章目录 例如&#xff1a;第一章 Python 机器学习入门之pandas的使用 文章目录 一、为什么需要内部类&#xff1f; 1.内部类的作用 2.内部类的使用场景 二、内部类的分类 1.成员内部类&#xff08;实例内部类&#xff09; 2.静态内部类 3.局部内部类 4.匿名内部类 总结 前…

ES如何提高准确率之【term-centric】

提高准确率的方法有很多&#xff0c;但是要在提高准确率的同时保证召回率往往比较困难&#xff0c;本文只介绍一种比较常见的情况。 问题场景 我们经常搜索内容&#xff0c;往往不止针对某个字段进行搜索&#xff0c;比如&#xff1a;标题、内容&#xff0c;往往都是一起搜索…

【卡塔尔世界杯数据可视化与新闻展示】

卡塔尔世界杯数据可视化与新闻展示 前言数据获取与处理可视化页面搭建功能实现新闻信息显示详情查看登录注册评论信息管理 创新点结语 前言 随着卡塔尔世界杯的临近&#xff0c;对于足球爱好者来说&#xff0c;对比赛的数据分析和新闻报道将成为关注的焦点。本文将介绍如何使用…

openmediavault debian linux安装配置企业私有网盘(三 )——raid5与btrfs文件系统无损原数据扩容

一、适用环境 1、企业自有物理专业服务器&#xff0c;一些敏感数据不外流时&#xff0c;使用openmediavault自建NAS系统&#xff1b; 2、在虚拟化环境中自建NAS系统&#xff0c;用于内网办公&#xff0c;或出差外网办公时&#xff0c;企业内的文件共享&#xff1b; 3、虚拟化环…

jmeter配置使用(mac)

前言 这篇文件就是一个笔记&#xff0c;非mac用户不用看了&#xff0c;我这是换了mac&#xff0c;要用jmeter的倒腾。 一、下载 二、使用步骤 1.解压 tgz格式的直接用tar命令就行 tar -zxvf 包名2.启动 一种是进入解压包的bin目录启动 这种方式启动的就是命令框不能关闭&am…

正则表达式详解

什么是正则表达式 正则表达式&#xff0c;又称规则表达式&#xff0c;通常被用来检索、替换那些符合某个模式(规则)的文本。 正则表达式是对字符串操作的一种逻辑公式&#xff0c;就是用事先定义好的一些特定字符、及这些特定字符的组合&#xff0c;组成一个"规则字符串…

小区生活污水处理需要哪些设备和工艺

在小区生活中&#xff0c;污水处理是一个非常重要的环节&#xff0c;它关乎到环境的保护和居民的生活质量。因此&#xff0c;了解小区生活污水处理所需要的设备和工艺是至关重要的。 首先&#xff0c;在小区生活污水处理中&#xff0c;需要用到的设备包括污水收集系统、初级沉淀…

【数据结构】——排序算法简答题模板

目录 一、内排序和外排序二、排序算法的稳定性三、插入排序&#xff08;一&#xff09;直接插入排序的步骤&#xff08;二&#xff09;直接插入排序的稳定性&#xff08;三&#xff09;折半插入排序的步骤&#xff08;四&#xff09;希尔排序的步骤 四、交换排序&#xff08;一…

多微信聚合聊天:一款简便的管理工具帮你摆脱微信繁琐之困!

在这个信息飞速传递的时代&#xff0c;微信已经成为了人们不可或缺的通信工具之一。然而&#xff0c;由于个人、工作等各种原因&#xff0c;我们可能会拥有多个微信号&#xff0c;而每天切换不同的账号进行聊天&#xff0c;无疑是一项繁琐的任务。 近日&#xff0c;我一位好友向…

互联网加竞赛 python+opencv+机器学习车牌识别

0 前言 &#x1f525; 优质竞赛项目系列&#xff0c;今天要分享的是 &#x1f6a9; 基于机器学习的车牌识别系统 &#x1f947;学长这里给一个题目综合评分(每项满分5分) 难度系数&#xff1a;4分工作量&#xff1a;4分创新点&#xff1a;3分 该项目较为新颖&#xff0c;适…

算法:程序员的数学读书笔记

目录 ​0的故事 ​一、按位计数法 二、不使用按位计数法的罗马数字 三、十进制转二进制​​​​​​​ ​四、0所起到的作用​​​​​​​ 逻辑 一、为何逻辑如此重要 二、兼顾完整性和排他性 三、逻辑 四、德摩根定律 五、真值表 六、文氏图 七、卡诺图 八、逻…

CorsInterceptor解决跨域问题

注解的方式 CrossOrigin(origins "http://localhost:8081", allowCredentials "true") 拦截器的方式 package com.rbynode.rbyaipro.interceptor;import org.springframework.web.servlet.HandlerInterceptor; import org.springframework.web.servl…

vector——OJ题

&#x1f4d8;北尘_&#xff1a;个人主页 &#x1f30e;个人专栏:《Linux操作系统》《经典算法试题 》《C》 《数据结构与算法》 ☀️走在路上&#xff0c;不忘来时的初心 文章目录 一、只出现一次的数字1、题目讲解2、思路讲解3、代码实现 二、杨辉三角1、题目讲解2、思路讲解…

前端没有死,前端开发框架:Bootstrap可以说一骑绝尘

再认识前端开发框架Bootstrap 当下最流行的前端开发框架Bootstrap,可大大简化网站开发过程,从而深受广大开发者的喜欢。本文总结了Bootstrap之所以广泛流传的11大原因。如果你还没有使用Twitter Bootstrap,建议你去了解一下。我也是最近才有所发现的,不过有更好的消息,在…