anaconda安装cuda

安装 CUDA 需要先安装 NVIDIA 显卡驱动程序。首先，你需要确认你的显卡是否支持 CUDA，并从 NVIDIA 官网上下载对应版本的 CUDA 驱动程序。然后，你可以按照以下步骤安装 CUDA。

1. 安装 Anaconda。

2. 在命令行中运行以下命令，创建一个新的 Python 环境并激活它：

conda create --name cuda_env python=3.7
conda activate cuda_env

3. 在命令行中运行以下命令，安装 CUDA 驱动程序和 CUDA 工具包：

conda install cudatoolkit

4. 在命令行中运行以下命令，安装 PyTorch 和 torchvision：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

注意：这里的 `cudatoolkit` 版本号需要与你下载的 CUDA 版本号一致。

5. 确认 CUDA 是否安装成功。在 Python 环境中运行以下代码：

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

如果输出结果为 True，则表示 CUDA 安装成功。

6. 最后，你需要配置 PyTorch 使用 CUDA。在 Python 环境中运行以下代码：

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

这将返回一个 PyTorch 设备对象，它将在 GPU 上运行代码（如果 CUDA 可用）或 CPU 上运行代码（如果 CUDA 不可用）。

希望这能帮到你！