You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

已装CUDA 11.8,RTX 3050 Ti Laptop GPU无法被PyTorch识别求助

解决PyTorch无法识别RTX 3050 Ti Laptop GPU的问题

1. 先验证基础环境

  • 打开终端运行nvidia-smi,确认显示的CUDA版本为11.8,且显卡正常识别
  • 运行nvcc --version,确认输出的CUDA版本与上述一致

2. 修正PyTorch与CUDA版本匹配问题

你当前安装的PyTorch绑定的是CUDA 11.7,但系统装的是CUDA 11.8,版本不匹配会导致GPU无法识别,按以下步骤操作:

  • 卸载现有PyTorch组件:
    conda uninstall pytorch torchvision torchaudio pytorch-cuda
    
  • 安装适配CUDA 11.8的PyTorch:
    conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
    
    若conda安装仍有问题,可改用pip安装:
    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
    

3. 验证GPU识别情况

运行以下Python代码确认:

import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name(0))

输出True、设备数1、"NVIDIA GeForce RTX 3050 Ti Laptop GPU"即为成功。

额外排查点

  • 确保操作时激活了正确的conda虚拟环境,避免在base环境中混用包
  • 检查笔记本显卡模式:打开NVIDIA控制面板,设置全局使用高性能NVIDIA处理器
  • 完成安装后重启电脑,确保驱动和配置完全生效

内容的提问来源于stack exchange,提问作者Alison

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.14 07:30:50