首先确认你是否有英伟达显卡,以及有驱动
按住 win 和 s键 输入cmd
打开这个
输入
nvidia-smi
如果有英伟达显卡却出现这个
是你没装驱动,或者看一下这个教程
Win11 ‘nvidia-smi‘ 不是内部或外部命令,也不是可运行的程序 或批处理文件。_nvidia-smi不是-CSDN博客
到英伟达官网选择需要的驱动就行,安装就行,只需要安装驱动,不需要安装cuda之类的东西
看图上红线的地方是目前驱动支持的最高cuda版本,记住这个版本号,后面有用
创建虚拟环境 -n 后面为环境名称我起的名字为cuda_12.1,这个可以自己指定, python=3.10指定安装的python版本为3.10
conda create -n xxx python=3.10
激活环境,activate 后跟的是你的环境名称
conda activate xxx
前面会有一个括号,这就代表你进入了名字叫cuda_12.1的环境
打开torch官网
按需选择,我用的是win11,我的最高支持的cuda版本是12.6,使用的是conda环境,所以我就这么选
复制这个
确保你进入了对应的虚拟环境,就是括号里面显示的,然后执行就行了。
假如你的cuda版本很低,点击这个,选择一个适合你的版本安装
等一会torch就安装好了,前提是你用的是校园网,Lingshui,如果是其他的网,需要改conda的下载源
安装他写的修改就行
anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
装好之后,验证一下,确保你选择了对应的conda环境
import torch
print(torch.cuda.is_available())
运行结果是True,torch安装成功。