新建mamba环境仅安装PyTorch CPU版，如何强制安装GPU版？

方法一：直接指定GPU版本后缀和匹配的CUDA toolkit #

在安装命令中通过pytorch=*=cudaXX明确指定要GPU版本的PyTorch，同时搭配对应版本的cudatoolkit。比如针对CUDA 11.6版本：

mamba install pytorch=*=cuda116 torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

如果需要匹配CUDA 11.3，把命令里的cuda116改成cuda113，cudatoolkit=11.6改成cudatoolkit=11.3即可。

方法二：先安装CUDA toolkit再安装PyTorch #

先创建包含对应版本CUDA toolkit的环境，让mamba自动匹配GPU版PyTorch：

1. 创建并激活环境：

mamba create -n your_env_name cudatoolkit=11.6
mamba activate your_env_name

2. 安装PyTorch套件：

mamba install pytorch torchvision torchaudio -c pytorch -c conda-forge

方法三：强制指定channel优先级避免依赖冲突 #

如果conda-forge等channel优先级过高导致拉取CPU版，可以用--override-channels强制只从指定channel获取包，确保PyTorch channel的优先级：

mamba install pytorch torchvision torchaudio cudatoolkit=11.6 --override-channels -c pytorch -c conda-forge

验证安装结果 #

激活环境后，运行以下代码验证GPU版本是否生效：

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

输出True则说明GPU版本安装成功。

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