3050显卡配置pytorch-gpu(cuda11.3版本)

本文介绍了如何在配备3050显卡的系统上,安装并配置CUDA11.3和CuDNN,以支持PyTorch的GPU版本。首先,通过NVIDIA控制面板检查显卡支持的CUDA最高版本。然后,从NVIDIA官方站点下载CUDA11.3和CuDNN。关键步骤是使用conda创建环境,并依据CUDA11.3的指令安装pytorch、torchvision和torchaudio。最后,确认环境正确配置,能够顺利运行深度学习模型。

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

1、显卡3050对应的CUDA最高支持版本可通过NVIDIA控制面板系统信息的组件里面查看可支持的最高版本。

2、通过以下网址,下载CUDA和CUDNN,我这里下载了11.3版本的,安装就按别人的教程来。

CUDA Toolkit Archive | NVIDIA Developer

 CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

3、最关键的一步!配置conda的环境。因为下载的为11.3的CUDA,所以必须根据cuda11.3的指令来下载,我这里用的命令为以下两条。

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 -c http://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/

 conda install cudatoolkit=11.3 -c http://mirrors.aliyun.com/anaconda/pkgs/main

3、用print(torch.cuda.is_available())检查是否完成

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值