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

news/2024/11/22 16:13:38/

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.ppmy.cn/news/209071.html

相关文章

《Apollo 智能驾驶进阶课程》二、 高精地图

1. 高精地图与自动驾驶的关系 1.1 高精地图与自动驾驶 L3级别以上才需要高精地图 1.2 什么是高精地图 1.3 高精地图与导航地图 1.4 高精地图-基础模块 高精地图与定位模块的关系 现在主流的自动驾驶的定位方案有两种:一种是基于点云,另一种是基于C…

从零搭建微服务-网关中心(一)

写在最前 如果这个项目让你有所收获,记得 Star 关注哦,这对我是非常不错的鼓励与支持。 源码地址:https://gitee.com/csps/mingyue 文档地址:https://gitee.com/csps/mingyue/wikis 新建 mingyue-gateway 在 【从零搭建微服务…

酷睿处理器(cpu)的型号分别代表什么?

酷睿处理器(cpu)的型号分别代表什么?如何从型号上来判断处理器的性能和好坏区分它们的档次? 1. Intel所有CPU都分为du高、中zhi、低三档,分别是i7、i5、i3,在这个档次后面dao还会跟着小型号。 2. 第一位代…

英特尔服务器主板型号哪里看,intel cpu型号参数如何查看

intel cpu型号参数如果想要查看下!那么我们要用什么方法去查看呢?下面由学习啦小编给你做出详细的intel cpu型号参数查看方法介绍!希望对你有帮助! intel cpu型号参数查看方法一 以Intel 酷睿i7 4710HQ处理器为例: intel是品牌名称; 酷睿i7,是系列名称;…

linux如何查看cpu型号

linux系统下!想要查看下cpu型号!用什么方法查看好呢下面由学习啦小编给你做出详细的linux查看cpu型号方法介绍!希望对你有帮助! linux查看cpu型号方法一 [rootsubvm ~]# less /proc/cpuinfo |grep model model : 4 model name : Intel Xeon™ CPU 3.00GHz model : 4 model…

计算机市场最主流的cpu,当前主流市场上有哪些CPU型号?

w404484343 在上个世纪,AMD和Intel CPU没有太大区别,因为当时的主流CPU是Intel的80x86 CPU。包括AMD在内的CPU也以x86命名。 现在AMD和Intel的CPU有很大的不同。 x86架构是Intel的专利,但是由于与AMD的早期合作,它被授权给AMD。 A…

Cpu 常见系列以及型号

Intel旗下 赛扬(Celeron)——桌面低端 奔腾(Pentium)—— 桌面中端 酷睿 (Core)——桌面高端 至强(Xeon)——服务器中端 安腾(Itanium)——服务器高端 凌动(Atom&#xf…

Intel CPU型号规格大全 (2007.08)

Intel CPU型号规格大全 (2007.08) 本文档的Copyleft归yfydz所有,使用GPL发布,可以自由拷贝,转载,转载时请保持文档的完整性, 严禁用于任何商业用途。 msn: yfydz_no1hotmail.com 来源: http://yfydz.cublog…