使用MINICONDA 创建pytorch环境

首先确认你是否有英伟达显卡,以及有驱动

按住 win 和 s键 输入cmd

打开这个

输入

nvidia-smi

如果有英伟达显卡却出现这个

是你没装驱动,或者看一下这个教程

Win11 ‘nvidia-smi‘ 不是内部或外部命令,也不是可运行的程序 或批处理文件。_nvidia-smi不是-CSDN博客

下载 NVIDIA 官方驱动 | NVIDIA

到英伟达官网选择需要的驱动就行,安装就行,只需要安装驱动,不需要安装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官网

PyTorch

按需选择,我用的是win11,我的最高支持的cuda版本是12.6,使用的是conda环境,所以我就这么选

复制这个

确保你进入了对应的虚拟环境,就是括号里面显示的,然后执行就行了。

假如你的cuda版本很低,点击这个,选择一个适合你的版本安装

等一会torch就安装好了,前提是你用的是校园网,Lingshui,如果是其他的网,需要改conda的下载源

安装他写的修改就行

anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

装好之后,验证一下,确保你选择了对应的conda环境

import torch

print(torch.cuda.is_available())

运行结果是True,torch安装成功。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值