使用Anaconda安装fastai/PyTorch时CUDA不可用的解决方法

问题根源 #

从你提供的conda list输出能明确看到，当前环境安装的是CPU版本的PyTorch（pytorch 1.12.1 cpu_py310h5e1f01c_0），这直接导致torch.cuda.is_available()返回False。fastai依赖PyTorch运行，只有当PyTorch是CUDA兼容版本时，才能调用你的RTX 3050 Ti GPU。

分步解决方案 #

1. 清理现有CPU版依赖 #

激活你的Anaconda环境后，卸载所有相关的CPU版包：

conda remove pytorch fastai torchvision torchaudio --all

2. 安装适配RTX 3050 Ti的CUDA版PyTorch与fastai #

RTX 3050 Ti支持CUDA 11.7及以上版本，优先安装带CUDA的PyTorch，再安装fastai（fastai会自动适配已安装的PyTorch版本）：

# 安装带CUDA 11.8的PyTorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
# 安装fastai
conda install -c fastchan fastai

3. 验证CUDA状态 #

安装完成后，在Python环境中执行以下代码验证：

import torch
print(torch.cuda.is_available())

若输出True，则说明CUDA已成功启用。

额外建议 #

• 确保NVIDIA显卡驱动为最新版本，驱动版本需适配CUDA 11.8（可通过NVIDIA控制面板检查更新）。
• 若旧环境存在顽固依赖冲突，建议创建全新的Anaconda环境：

  conda create -n fastai-gpu python=3.10
  conda activate fastai-gpu
  

  之后再执行上述安装步骤。

内容的提问来源于stack exchange，提问作者Shwetansh Singh