CUDA12.2 conda安装torch

本文详细描述了一个新手如何通过更换默认源、安装特定版本的CUDA并使用conda在Anaconda环境中安装PyTorch,包括创建虚拟环境、确认CUDA版本和执行安装步骤的过程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

写在前面:这是一个完整的利用镜像安装torch的方法。本人是小白,所以写的很基础很详细。

我自己安装的时候参考了下列帖子,非常感谢大佬们的分享:

conda安装pytorch出现Solving environment: unsuccessful initial attempt using frozen solve.-优快云博客

【真正解决】Conda安装卡在solving environment这一步!-优快云博客 

基本上看了这俩帖子就可以自己安装了,不过我还是想详细的完整的写一下安装过程(毕竟我花了3天才装好,踩了好多坑)。

首先你的电脑要装有anaconda,并建立好一个虚拟环境。我们将在这个虚拟环境里安装torch。

一、创建虚拟环境

首先,打开anaconda prompt,并激活/创建一个虚拟环境。

创建虚拟环境(这里虚拟环境我命名为test,后面python=x.x指的是python版本,这个可以自己定义):

conda create -n test python=x.x

 二、换源

好的,最重要的时刻来了。有些人之前可能为了提高下载速度装了很多源,正如开头参考贴中所说,这些杂乱的源可能是torch安装不成功的原因。

所以,首先我们需要恢复默认源:

conda config --remove-key channels

在此之后,我们需要安装新的镜像源(from开头参考贴):

conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/main
 
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/r
 
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/msys2
 
conda config --set show_channel_urls yes

然后,有一个非常重要的动作!打开你的C盘,找到C盘下user下的.condarc文件,并用记事本打开它,删掉-default(下图是我自己删过的):

三、确认本机CUDA版本&找到torch官网安装命令

CUDA确认方法:同时按住windows+r键,在弹出的框里输入cmd,回车。在黑色命令窗口输入nvidia-smi,就会看到黑色的命令窗口里出现了一些信息,如下图。可以看到第一行右侧有个CUDA Version:12.2,这就是该电脑的CUDA版本。

(解释:确认CUDA以确定接下来torch安装哪个版本)

torch安装命令:搜索torch,进入torch官网(进不去可能需要科学上网),然后我们下拉,可以看到如下界面,安装命令我贴在下面了(其中,stable是稳定版本;window是系统不多说了;conda是,,,我理解是安装方式的意思,不知道对不对,反正选它就行;python是语言;最后这个Compute platform一般是要选和自己电脑CUDA版本一致的才对,但是12.2目前还没有对应的,因此先装一个12.1的,据说是可以用。):

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

四、安装torch

 激活虚拟环境(test为虚拟环境名称):

conda activate test

然后把上述安装命令直接贴上去,不需要删除任何东西,直接回车运行。等电脑自己装,大概十几分钟,应该就能装好。

欢迎交流。

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值