ubuntu22.04 安装torch
时间: 2024-10-06 12:03:56 浏览: 43
在Ubuntu 22.04上安装PyTorch(一个流行的深度学习库),你可以按照以下步骤操作:
1. **更新系统**:
```sh
sudo apt update
sudo apt upgrade
```
2. **安装依赖项**:
对于Python 3.6+,你需要安装一些必要的包:
```sh
sudo apt install -y software-properties-common python3-pip
```
3. **添加官方PPA(Personal Package Archive)**,这是获取最新PyTorch版本的一种常见方法:
```sh
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
sudo apt-get update
```
4. **安装GCC 9**(对于某些特定版本的PyTorch,如1.8+,需要这个编译器):
```sh
sudo apt install g++-9
```
5. **安装Python 3(如果你还没有安装)**:
```sh
sudo apt install python3
```
6. **安装PyTorch**:
如果你需要CPU版本:
```sh
sudo apt install torch torchvision
```
如果需要GPU支持(CUDA和cuDNN),首先确认你的显卡兼容并且已经安装了CUDA:
- 查看是否已安装CUDA:
```sh
nvidia-smi
```
- 安装相应的cuda和cudnn:
```sh
sudo apt-get install cuda cudnn
```
然后安装PyTorch:
```sh
CUDA_HOME=/usr/local/cuda pip3 install torch torchvision -f https://download.pytorch.org/whl/cu11[0-9].html
```
7. **安装完成后的验证**:
```sh
python3 -c "import torch; print(torch.__version__)"
```
注意:以上步骤可能会因Linux发行版的具体细节略有差异。如果遇到错误,可能需要查阅更多详细的文档或在线教程。
阅读全文