PyTorch在Ubuntu上的配置方法

在Ubuntu上配置PyTorch可以分为几个步骤，包括安装必要的软件包、设置CUDA和cuDNN、以及安装PyTorch本身。以下是详细的步骤指南：

1. 安装Ubuntu系统

• 下载并制作Ubuntu 20.04 LTS的启动盘。
• 使用启动盘安装Ubuntu系统。

2. 配置Ubuntu环境和软件包

• 设置root密码。
• 更换Ubuntu的国内源以加快软件包的下载速度。

3. 安装NVIDIA显卡驱动

• 使用命令 nvidia-smi 查看显卡信息。
• 安装推荐的NVIDIA显卡驱动。

4. 安装CUDA Toolkit

• 访问NVIDIA官方网站，下载适用于Ubuntu版本的CUDA Toolkit。
• 按照官方指南安装CUDA。

5. 安装cuDNN

• 访问NVIDIA官方网站，下载与CUDA版本相匹配的cuDNN库。
• 解压并配置环境变量。

6. 安装Anaconda或Miniconda

• 下载并运行Anaconda或Miniconda的安装脚本进行安装。
• 创建并激活一个Conda虚拟环境。

7. 安装PyTorch

• CPU版本：

conda install pytorch torchvision cpuonly -c pytorch

• GPU版本：

conda install pytorch torchvision cudatoolkit=12.1 -c pytorch -c nvidia

注意：在安装GPU版本的PyTorch之前，确保你的CUDA和cuDNN版本与PyTorch的要求相匹配。

8. 验证安装

• 在Python环境中导入PyTorch，并检查CUDA是否可用。

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

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

以上步骤应该能够帮助你在Ubuntu上成功配置PyTorch环境。如果在安装过程中遇到问题，可以参考PyTorch的官方文档或相关社区论坛寻求帮助。