如何在conda环境下下载和CUDA11.4匹配的torch

在Conda环境中下载与CUDA 11.4相匹配的PyTorch，你需要按照以下步骤操作：

1. **创建新环境**:
首先，打开终端或命令提示符，确保已经激活了Conda环境。如果没有，可以输入 `conda create -n my_torch_cuda114 python=3.7` 创建一个新的名为`my_torch_cuda114` 的环境，这里假设Python版本为3.7。替换实际需要的Python版本。

2. **激活环境**:
输入 `conda activate my_torch_cuda114` 激活刚创建的环境。

3. **安装CUDA和cuDNN**:
CUDA 11.4对应的cudnn版本通常是cuDNN 8.0。你可以使用以下命令安装CUDA和cuDNN:

   conda install cudatoolkit=11.4 cudnn=8.0.5 -c anaconda

确保将上述版本号替换为你实际需要的版本。

4. **安装PyTorch**:
使用Conda安装特定版本的PyTorch，它通常会自动与所选的CUDA版本兼容。运行：

   conda install pytorch torchvision torchaudio -c pytorch

如果你遇到找不到合适版本的问题，可以在`pytorch`后面加上`cuda=11.4`，如 `conda install pytorch torchvision torchaudio -c pytorch cuda=11.4`。

5. **验证安装**:
安装完成后，可以通过`python -m torch.cuda.is_available()`检查是否成功加载了GPU加速。如果一切正常，应该返回`True`。