raise AssertionError(“Torch not compiled with CUDA enabled“)

news2024/11/26 12:33:37

1、运行代码可知,当前cuda不可用。

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

2、打开power shell or cmd,输入nvidia-smi命令,检查当前英伟达显卡信息。

可知当前驱动版本512.78,支持的cuda最高版本为11.6,而pytorch 2.0.0需要cuda11.7或11.8版本。

3、下载最新版本GeForce(非最新版本驱动会更新失败),安装,并在里面更新驱动:NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

当前驱动版本已更新至535.98,支持的cuda最高版本已经变为12.2。

4、接下来更新cuda

运行 nvcc -V 可知,当前cuda版本为 11.1。

5、下载 cuda11.8:CUDA Toolkit Archive | NVIDIA Developer

选择自定义安装

C盘资源紧张的话,可以换个非C盘的路径。(即使这样,安装后C盘也会占用5~6个G)

 

安装完之后在环境变量->系统变量->Path里面添加CUDA路径。

6、运行 nvcc -V 可知,当前cuda版本已更新为 11.8。

 7、接下来安装cudnn

下载cudnn:cuDNN Archive | NVIDIA Developer

下载前要先注册信息,不算特别麻烦,勾几个√就行。

在for CUDA 11.x 里面随便选一个版本,我选的是最新版本

 下载完成后,将压缩包解压,将 binincludelib 3个文件夹复制到cuda的安装目录v11.8下。

8、安装pytorch2.0

进入pytorch官网:PyTorch

conda create -n pytorch2.0 python==3.9
conda activate pytorch2.0
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

我这里是在conda上创建的虚拟环境,不过还是用pip命令安装比较方便。

(想要自己下载的话,可以访问:https://download.pytorch.org/whl/torch/)

9、检查 cuda 是否可用

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

终于成功啦!主要是下载、安装花时间,一步一步操作下来没什么技术难度,也没有遇到难搞的问题。

总结以下经验:cuda 切换版本以后,pytorch2.0 的环境要重新搭才行。因为我之前的pytorch2.0.1用不了,重新建conda虚拟环境安装 pytorch2.0.1 就可以用了。

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

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

相关文章

时间序列教程 六、深度学习与时间序列分析结合

一、深度学习方法 与传统的时间序列预测模型相比,神经网络有以下几个好处: 1、自动学习如何将趋势、季节性和自相关等系列特征纳入预测。 2、能够捕捉非常复杂的模式。 3、可以同时建模许多相关的系列,而不是单独处理每个系列。 但是神经网络有一些劣势: 1、模型的构建可能…

PyCharm开发工具的安装与使用

PyCharm集成开发工具(IDE),是当下全球python开发者,使用最频繁的工具软件。绝大多数的python程序,都是在PyCharm工具内完成的开发。 1.先进行下载并安装它 下载官网地址:https://www.jetbrains.com/pycharm/download/#sectionwindows 宝子…

SpringBoot自定义拦截器实现权限过滤功能(基于责任链模式)

前段时间写过一篇关于自定义拦截器实现权限过滤的文章,当时是用了自定义mybatis拦截器实现的:SpringBoot自定义Mybatis拦截器实现扩展功能(比如数据权限控制)。最近学习设计模式发现可以用责任链模式实现权限过滤,因此本篇采用责任链模式设计…

Docker(概述、安装、配置、镜像操作)

一、docker是什么? docker是一种go语言开发的应用容器引擎,运行容器里的应用。docker是用来管理容器和镜像的一种工具。 容器引擎:docker、rocket、podman、containerd 容器与虚拟机的区别 容器:所有容器共享宿主机内核。使用…

【手撕Spring源码】AOP

文章目录 AOP 实现之 ajc 编译器AOP 实现之 agent 类加载AOP 实现之 proxyJDK代理CGLIB代理JDK动态代理进阶CGLIB代理进阶MethodProxy JDK 和 CGLIB 在 Spring 中的统一切点匹配从 Aspect 到 Advisor通知转换调用链执行静态通知调用动态通知调用 AOP 底层实现方式之一是代理&am…

Java/Compose Desktop项目中进行python调用

写在前面 开发compose desktop项目爬网站时遇到验证码处理不方便需要借助python庞大的处理能力,这时需要再项目中调用python去执行获取结果,这里记录一下使用过程。 本次开发记录基于:python 3.9,compose 1.3 java17 工具&#x…

2年测试我迷茫了,软件测试大佬都会哪些技能?我的测试进阶之路...

目录:导读 前言一、Python编程入门到精通二、接口自动化项目实战三、Web自动化项目实战四、App自动化项目实战五、一线大厂简历六、测试开发DevOps体系七、常用自动化测试工具八、JMeter性能测试九、总结(尾部小惊喜) 前言 Python自动化测试&…

【MCS-51单片机汇编语言】期末复习总结④——求定时器初值(题型四)

文章目录 重要公式T~机器~ 12 / ∫~晶振~(2^n^ - X) * T~机器~ T~定时~ 工作方式寄存器TMOD常考题型例题1题解方式0方式1 关于定时器的常考题目为已知晶振 ∫ 、定时时间,求定时器初值。 重要公式 T机器 12 / ∫晶振 (2n - X) * T机器 T定时 其中n为定时器位数…

线性代数2:矩阵(1)

目录 矩阵: 矩阵的定义: 0矩阵 方阵 同型矩阵: 矩阵相等的判定条件 矩阵的三则运算: 乘法的适用条件 矩阵与常数的乘法: 矩阵的乘法: 矩阵的乘法法则: Note1: Note2&…

【数据库】表数据delete了,表文件大小不变

背景 在本周的时候,接到了短信数据空间报警短信,提示的是磁盘空间占用80以上,而这个数据库总体的存储量一共100G,商量之后决定在不升配置的前提下,删除一些不需要的数据表。比如针对A表删除1000W数据。但是和DBA沟通后…

FAST-LIO2论文阅读

目录 迭代扩展卡尔曼滤波增量式kd-tree(ikd-tree)增量式维护示意图ikd-tree基本结构与构建ikd-tree的增量更新(Incremental Updates)逐点插入与地图下采样使用lazy labels的盒式删除属性更新 ikd-tree重平衡平衡准则重建及并行重建…

SMTP简单邮件传输协议(C/C++ 发送电子邮件)

SMTP是用于通过Internet发送电子邮件的协议。电子邮件客户端(如Microsoft Outlook或macOS Mail应用程序)使用SMTP连接到邮件服务器并发送电子邮件。邮件服务器还使用SMTP将邮件从一个邮件服务器交换到另一个。它不用于从服务器下载电子邮件;相…

jmeter安装及使用

jmeter安装及使用 一、说明二、安装2.1 目录结构 三、使用3.1 运行jmeter3.2 设置语言3.3 设置线程组3.3.1 设置压测请求3.3.2 设置汇总报告3.3.3 设置结果树 3.4 开始压测 四、导出执行报告4.1 保存配置4.2 执行命令4.3 生成报告常见问题 一、说明 最近需要对项目接口进行压测…

vue简单实现一个类似微信左右滑动更多功能

1、需求背景 产品需要在购物车加一个左右滑动更多的功能,由于是PC端,大致扫描了下使用的UI库,貌似没有单独提供此类组件,反正有时间,就自己造一个轮子试试 2、先看效果 大致有一个橡皮筋的效果,可能没那…

分布式锁方案学习

很久没有写文章了,前些天的面试被问到了分布式锁的解决方案,回答的比较简单,只知道Redis,Mysql,Zookeeper能够作为分布式锁应用,今天就来详细的学习一下这三种分布式锁的设计思想及原理。 能够来看这篇文章…

05WEB系统的通信原理图

WEB系统的通信原理 名称作用URL统一资源定位符, 例如:http://www.baidu.com域名在https://www.baidu.com/这个网址中www.baidu.com 是一个域名IP地址计算机在网络当中的一个身份证号, 在同一个网络当中IP地址是唯一的, 有了IP地址两台计算机直接才能建立连接通信端口号一个计算…

如何让你的汇报更有说服力?数据监控是关键!

第5讲中玩过一个扫雷游戏,目标是排除计划中的“延期地雷”,但是,总有些“雷”防不胜防。我们在做计划的时候,明明已经想得非常周全了,可是,真正开工几天之后才发现,很多事情并没有那么简单。 1…

4-1 活动安排问题

1.什么是贪心算法 我的理解: 贪心算法是一种常用的问题求解方法,它在每个步骤上都选择当前看起来最优的解,而不考虑整体的最优解。简单来说,贪心算法采取局部最优的决策,希望通过每个局部最优解的选择,最终…

网络安全面试题大全(整理版)500+面试题附答案详解,最全面详细,看完稳了

前言 随着国家政策的扶持,网络安全行业也越来越为大众所熟知,想要进入到网络安全行业的人也越来越多。 为了拿到心仪的Offer之外,除了学好网络安全知识以外,还要应对好企业的面试。 作为一个安全老鸟,工作这么多年&…

全网最全的网络安全技术栈内容梳理(持续更新中)

前言 本文篇幅比较长~~耐心看完哦~ 网络安全真的那么好吗 据我了解现在我国网络安全人才缺口相当大,预计在2023年这方面人才缺口达到327万,我每年这方面的大学生才2W多。现在各政企都在发展数字化变革,对网络安全方面人才也是垂涎若渴&…