Ubuntu22.04安装pytorch cuda python
时间: 2025-04-17 14:38:59 浏览: 33
### 安装支持 CUDA 的 PyTorch
为了在 Ubuntu 22.04 上成功安装支持 CUDA 的 PyTorch,需遵循一系列特定的操作流程来确保兼容性和稳定性。
#### 准备工作
确认系统已正确配置 NVIDIA 驱动程序以及所需的依赖项。这一步骤对于后续 CUDA 和 PyTorch 的正常运行至关重要[^2]。
#### 安装 CUDA 工具包
通过官方渠道获取并安装最新版的 CUDA Toolkit 是必要的前置条件之一。具体命令如下所示:
```bash
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda
```
完成上述步骤后重启计算机以使更改生效[^3]。
#### 安装 cuDNN 库
cuDNN 对于加速深度学习模型训练非常重要。可以通过下载对应的 `.deb` 文件来进行本地安装:
```bash
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.6.0.163_1.0-1_amd64.deb
```
之后再次更新软件源列表,并按照提示完成剩余部分的设置[^1]。
#### 获取适合当前环境的 PyTorch 版本
访问 [PyTorch官方网站](https://pytorch.org/get-started/locally/) ,根据个人需求选择合适的 PyTorch 构建选项(包括操作系统、Python 版本、CUDA 版本)。通常推荐使用 pip 或 conda 来简化整个过程[^4]。
例如,如果选择了 Python 3.x 和 CUDA 11.y,则可以执行以下命令进行安装:
```bash
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
```
注意替换 `cu117` 为实际使用的 CUDA 版本号字符串。
#### 测试安装成果
最后验证新安装的 PyTorch 是否能够识别 GPU 设备:
```python
import torch
print(torch.cuda.is_available())
print(torch.__version__)
```
当输出显示 True 并打印出正确的 PyTorch 版本信息时即表示安装成功。
阅读全文
相关推荐


















