ubuntu20.04安装pytorch 1.10
时间: 2023-08-25 21:08:21 浏览: 133
升级pip到20.0.2+用本地安装包安装pytorch
在 Ubuntu 20.04 中安装 PyTorch 的步骤如下:
1. 安装 Anaconda(如果您还没有安装),可以在终端中使用以下命令安装:
```
wget https://repo.anaconda.com/archive/Anaconda3-2021.07-Linux-x86_64.sh
bash Anaconda3-2021.07-Linux-x86_64.sh
```
2. 创建一个新的 Anaconda 环境,如:
```
conda create -n pytorchenv python=3.8
```
3. 激活该环境
```
conda activate pytorchenv
```
4. 安装 PyTorch
```
conda install pytorch torchvision -c pytorch
```
5. 安装 torchaudio
```
conda install torchaudio -c pytorch
```
6. 若需要 cudatoolkit 支持,在安装 pytorch时添加 cudatoolkit 版本, 例如安装 CUDA 11.0 版本:
```
conda install pytorch torchvision cudatoolkit=11.0 -c pytorch
```
7. 验证安装是否成功,可以在终端中使用以下命令
```
python -c "import torch;print(torch.__version__)"
```
若看到相应的版本号,说明安装成功。
阅读全文