当前位置：首页 > article >正文

Anaconda 下个人环境的快速安装指南：支持 GPU 运算的 PyTorch 环境

article 2025/2/7 8:22:12

Anaconda 下个人环境的安装指南：支持 GPU 运算的 PyTorch 环境

这个安装指南将指导你如何使用 Anaconda 创建一个支持 GPU 运算的 PyTorch 环境。我们将涵盖以下步骤：

查询显卡驱动和 CUDA 版本
创建 yml 文件并通过 Conda 创建环境
安装 PyTorch 和相关库
验证 PyTorch 能否使用 GPU

步骤 1：查询显卡驱动和 CUDA 版本

检查 NVIDIA 驱动：
- Windows：打开命令提示符（cmd）或 Anaconda Prompt，运行：
```
nvidia-smi
```
- Linux/Mac：打开终端，运行：
```
nvidia-smi
```
这会显示当前安装的 NVIDIA 驱动 版本和 CUDA 版本。确保你的驱动版本与 CUDA 兼容。如果没有正确安装驱动，请前往 NVIDIA 官网下载并安装最新驱动。

确认 CUDA 版本：
运行 nvidia-smi 后，你可以看到类似如下信息：

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.89       Driver Version: 460.89       CUDA Version: 11.3     |
+-----------------------------------------------------------------------------+

这里的 CUDA Version: 11.3 就是你当前使用的 CUDA 版本。

步骤 2：创建 `yml` 文件并通过 Conda 创建环境

创建 environment.yml 文件：
使用文本编辑器（如 Notepad、VS Code 或任何编辑器），创建一个名为 environment.yml 的文件，并添加以下内容：

name: pytorch_gpu_env  # 环境名称，可以根据需要修改
channels:
  - pytorch  # PyTorch 官方源
  - nvidia   # NVIDIA 官方源，提供 CUDA 相关依赖
  - defaults # 默认 Conda 源
dependencies:
  - pytorch=1.12.0  # 选择与 CUDA 11.3 兼容的 PyTorch 版本
  - torchvision=0.13.0  # 与 PyTorch 配套的图像处理库
  - torchaudio=0.12.0  # 与 PyTorch 配套的音频处理库
  - cudatoolkit=11.3  # 确保安装与 GPU 驱动兼容的 CUDA 版本
  - numpy  # 数值计算库
  - pip
  - pip:
    - matplotlib  # 可视化库（如果需要的话）
    - scikit-learn  # 机器学习库（如果需要的话）

在这个 yml 文件中：

pytorch=1.12.0 表示使用与 CUDA 11.3 兼容的 PyTorch 版本。
cudatoolkit=11.3 确保安装与 CUDA 11.3 驱动版本兼容的 CUDA 工具包。

创建 Conda 环境：
打开 Anaconda Prompt 或终端，进入你保存 environment.yml 文件的目录，并运行以下命令：
```
conda env create -f environment.yml
```
Conda 会自动下载并安装所需的依赖，包括 PyTorch、CUDA 和其他指定的库。
激活环境：
创建环境后，激活你的 Conda 环境：
```
conda activate pytorch_gpu_env
```
现在，你的环境已经准备好了，接下来可以安装其他依赖或运行代码。

步骤 3：安装 PyTorch 和相关库

在 Conda 环境中，安装 PyTorch 和相关的库（如果你没有在 yml 文件中指定）：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

如果你已经在 yml 文件中指定了这些库，则跳过此步骤。

步骤 4：验证 PyTorch 是否支持 GPU

测试 PyTorch 是否能够访问 GPU：
在 Python 环境中运行以下代码，验证是否能够成功使用 GPU：

import torch

# 查看 PyTorch 版本
print(torch.__version__)

# 检查是否有可用的 GPU
print(torch.cuda.is_available())  # 如果返回 True，表示 GPU 可用

# 获取 GPU 的名称
if torch.cuda.is_available():
    print(torch.cuda.get_device_name(0))  # 获取第一个 GPU 的名称

如果 torch.cuda.is_available() 返回 True，且 torch.cuda.get_device_name(0) 返回你的 GPU 名称，说明 PyTorch 能够正确访问你的 GPU。

运行简单的 GPU 计算测试：
你可以运行一个简单的 GPU 计算测试，验证 GPU 运算是否正常：
```
# 创建一个张量，并将其移动到 GPU 上
x = torch.rand(5, 5).cuda()
print(x)
```
如果没有错误，并且输出是一个张量，说明 GPU 正常工作。

附加提示

如果遇到问题：
- 如果 torch.cuda.is_available() 返回 False，请确保显卡驱动和 CUDA 版本兼容，且驱动已经正确安装。
- 如果你使用的是 NVIDIA 显卡，确保已经安装了对应的 CUDA 工具包。
更新 PyTorch：
- 如果你需要更新 PyTorch，可以使用以下命令：
```
conda update pytorch
```