ubuntu中pytorch安装
时间: 2024-12-26 11:20:05 浏览: 6
### 安装 PyTorch 的方法
对于希望在 Ubuntu 上安装 PyTorch 而不依赖 Anaconda 的用户来说,可以采用 `pip` 工具来完成这一过程。考虑到不同硬件配置的需求,在安装前需确认目标 CUDA 版本以匹配本地 GPU 驱动情况[^2]。
#### 使用 Pip 安装特定版本的 PyTorch 和 TorchVision
针对仅配备 CPU 或者偏好纯 CPU 运算环境的情况,可以通过指定 URL 来获取稳定版次:
```bash
sudo apt install python3-pip
pip3 install torch==1.9.1+cpu torchvision==0.10.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
```
上述命令会下载并安装适用于 Linux 平台、基于 Python 3.x 架构下的 PyTorch 及其配套库——TorchVision,并且特别指定了只含 CPU 支持而不涉及任何 GPU 加速组件的版本号[^3]。
如果计算机配备了 NVIDIA 显卡并且已正确设置了相应的驱动程序以及兼容版本的 CUDA Toolkit,则可以选择带有 GPU 支持的 PyTorch 发行版。此时应访问官方网页查询最新推荐组合,依据个人需求挑选合适的二进制包进行部署。
#### 创建软链接以便全局调用 Conda (可选)
当遇到无法正常使用 `conda` 命令的情形时,一种解决方案是在 `/usr/local/bin/` 中建立指向实际路径(如 anaconda3/bin/ 下)的符号链接:
```bash
ln -s /path/to/anaconda3/bin/conda /usr/local/bin/
```
不过需要注意的是此操作并非强制要求,除非确实计划频繁运用到 Conda 功能或是已有现成的 Anaconda 环境存在。
阅读全文