ubuntu 22 pytorch 安装
时间: 2025-03-26 20:31:49 浏览: 28
### 安装 PyTorch 的准备工作
为了确保顺利安装 PyTorch,在 Ubuntu 22.04 LTS 上建议先更新系统的软件包列表并升级已有的软件包:
```bash
sudo apt update && sudo apt upgrade -y
```
接着,推荐安装 Anaconda 或 Miniconda 来管理 Python 环境以及依赖项。通过 Conda 创建独立的虚拟环境有助于避免不同项目之间的库版本冲突。
对于 GPU 支持的需求,需确认系统已经正确配置了 NVIDIA 驱动程序和 CUDA 工具包[^1]。
### 使用 Conda 安装 PyTorch
创建一个新的 conda 虚拟环境 `myenv` 并激活它:
```bash
conda create --name myenv python=3.9
conda activate myenv
```
之后可以通过官方渠道获取最新的稳定版 PyTorch。考虑到国内网络状况可能导致下载缓慢的问题,可选用清华大学开源软件镜像站作为加速手段之一[^2]。
执行如下命令来安装带有 CUDA 加速功能的 PyTorch 版本(假设使用的是 CUDA 11.7):
```bash
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
```
上述操作完成后即可完成 PyTorch 的基本安装流程。
### 验证 PyTorch 是否成功安装
要在新环境中测试 PyTorch 及其对 GPU 的支持情况,可以在终端内启动 Python 解释器,并运行以下代码片段来进行验证:
```python
import torch
print(torch.__version__)
if torch.cuda.is_available():
print('CUDA is available')
else:
print('No CUDA support found.')
```
当输出显示有可用的 CUDA 设备时即表明 PyTorch 成功安装并且能够利用到 GPU 计算资源。
阅读全文
相关推荐

















