ubuntu中下载torch
时间: 2024-10-22 13:26:18 浏览: 25
Ubuntu中安装GPU版本pytorch
5星 · 资源好评率100%
在Ubuntu系统中安装Torch(通常指PyTorch,一个开源的机器学习库)可以通过包管理器`apt`或者从源码编译安装。以下是两种常见的安装方法:
1. **通过`apt`包管理器安装** (推荐对于初学者)
- 首先确保你的系统是最新的,并且已经添加了PyPI的PPA(Personal Package Archive):
```bash
sudo apt update
sudo apt install python3-pip
sudo add-apt-repository ppa:pytorch/ppa
```
- 然后安装torch:
```bash
sudo apt-get install torch torchvision
```
2. **从源代码编译安装(适用于需要最新版本或特定需求的情况)**
- 先安装必要的依赖项,如CUDA、cuDNN等,如果你有GPU,可以参考官方文档:
- https://pytorch.org/docs/stable/installation.html?highlight=torch%20install#linux
- 完成依赖安装后,访问PyTorch官网获取最新版本的源代码并解压:
```
git clone https://github.com/pytorch/pytorch.git
cd pytorch
```
- 运行编译安装脚本:
```bash
export CUDA_HOME=/usr/local/cuda
./build.py --cuda-version=x.x --cudnn-version=x.x --config=release
```
(x.x替换为实际的CUDA和cuDNN版本)
安装完成后,你可以通过`python3 -c "import torch"`来验证Torch是否成功安装。
阅读全文