ubuntu安装libtorch
时间: 2024-08-16 22:00:31 浏览: 149
0296-极智开发-解读ubuntu安装libtorch的方法
在Ubuntu系统上安装libtorch,你需要先确保已经安装了CUDA和cuDNN,并且通过包管理器更新软件包。以下是基本步骤:
1. **安装依赖**:
- 打开终端并运行:
```
sudo apt-get update
sudo apt-get install build-essential cmake pkg-config libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-dev protobuf-compiler libopenblas-dev liblapack-dev gfortran
```
2. **添加官方apt源**:
如果你是CUDA 11.x用户,可以添加以下PPA源:
```
sudo add-apt-repository ppa:pytorch/vision-gpu
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
```
对于CUDA 10.x,使用以下PPA:
```
sudo add-10
```
3. **更新和安装**:
```
sudo apt-get update
sudo apt-get install torch torchvision python3-torch
```
4. **检查安装**:
安装完成后,你可以通过Python命令行测试安装是否成功:
```
python3 -c "import torch; print(torch.__version__)"
```
如果你需要特定版本的libtorch,如nightly版,可以参考PyTorch官网的安装指南(https://pytorch.org/get-started/locally/)。
阅读全文