CUDA 12.2 配置pytorch简单版(Windows版本,GPU)

之前一直使用的虚拟机,但是等到跑深度学习模型的时候才发现,cuda和pytorch不匹配,报错如下:AssertionError: Torch not compiled with CUDA enabled。并且更致命的事情是,虚拟机无法调用主机中的GPU,故直接下载Windows版本的。

电脑的显卡是4060的,cuda版本是12.2,查看的方法可以从我之前的博文获取。但是pytorch官网(PyTorch)目前只更新到cuda12.1的。

 为此,查了好多方法,其中有一个方法就是将cuda降级,但是博主觉得实在是太麻烦了。便直接下载了最新的pytorch,意外的发现可以匹配,并且完全可以用。故分享方法如下:

首先,下载Anaconda,并且按照步骤进行安装即可(选择的文件夹需要为空文件夹,不然会报错)。安装完之后,在开始的菜单里面,点击下图圈出来的,打开类似于linux系统的终端:

 打开之后,创建环境:

conda create -n pytorch python=3.8

pytorch是我自己定义的名字,可以根据自己的选择进行更改。

之后输入

conda activate torch

出现如下图所示的便为激活成功:

然后便可以去官网下载pytorch,建议下载stable版本的。

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

由于网站是国外的原因,下载速度是很感人的,因此可以通过以下命令添加镜像源,下载速度会快一些(比不添加快一点只是,不知道怎么回事,和虚拟机上面的速度没法比)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
 

然后再把上面的pytorch下载链接改为如下的:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1

在solving environment如果提示比如:EnvironmentNotWritableError: The current user does not have write permissions to the target environm。一般是没有足够的权限,解决方法是,在打开上面的Anaconda Powershell Prompt的时候,选择以管理员身份运行即可。之后问题便被成功解决,最后,看cuda和pytorch是否匹配:

首先python,然后输入;

import torch

然后,

torch.cuda.is_available()

如果返回True,便代表安装成功

  • 36
    点赞
  • 57
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值