ubuntu cuda11.8安装 pytorch
时间: 2025-01-31 22:27:37 浏览: 108
### Ubuntu 上安装 CUDA 11.8 并配置 PyTorch
#### 下载 NVIDIA 驱动程序
对于 Ubuntu 系统,推荐的方法是从官方渠道获取最新的驱动程序。可以通过系统的软件更新功能来完成这一操作,在 Ubuntu 的 Software & Updates 工具中启用 Proprietary drivers for your GPU 即可获得最新版本的 NVIDIA 显卡驱动[^2]。
#### 获取 CUDA 11.8 安装文件
访问 NVIDIA 开发者网站下载适用于 Linux 的 CUDA Toolkit 11.8 版本。如果遇到网络问题无法通过 `wget` 命令在线下载,则可以选择手动从网页端下载安装包并上传到目标机器上执行安装过程[^4]。
```bash
# 手动下载链接(需浏览器打开)
https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
```
#### 运行 CUDA Installer
确保已经关闭所有不必要的应用程序和服务,并按照提示运行安装脚本:
```bash
sudo sh cuda_11.8.0_520.61.05_linux.run
```
此过程中会询问关于是否要安装第三方组件等问题,请根据实际情况作出选择[^3]。
#### 设置环境变量
编辑用户的 `.bashrc` 文件以添加必要的路径设置,使系统能够识别新安装的 CUDA 库位置:
```bash
echo 'export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
source ~/.bashrc
```
#### 安装兼容版 PyTorch
考虑到当前 PyTorch 构建对 CUDA 支持的最大版本为 11.8,建议采用 pip 或 conda 方式安装特定于该版本的 PyTorch 发行版。可以参照官方网站给出的具体指令进行操作[^1]。
```python
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
```
阅读全文
相关推荐


















