ubuntu安装pytorch\
时间: 2023-09-06 17:13:35 浏览: 112
在 Ubuntu 上安装 PyTorch 有多种方式,以下是其中两种常用的方法:
## 方法一:使用 pip 安装
1. 确保你的 Python 版本是 3.6 或以上。
2. 安装 pip:
```
sudo apt-get update && sudo apt-get install python3-pip
```
3. 安装 PyTorch:
```
pip3 install torch torchvision
```
4. 验证 PyTorch 是否安装成功:
```
python3 -c "import torch; print(torch.__version__)"
```
如果安装成功,将会输出 PyTorch 的版本号。
## 方法二:使用 conda 安装
1. 安装 Anaconda:在 [官网](https://www.anaconda.com/products/individual) 下载适合你系统的 Anaconda 安装包,然后安装。
2. 创建一个新的 conda 环境:
```
conda create -n myenv python=3.6
```
3. 激活 conda 环境:
```
conda activate myenv
```
4. 安装 PyTorch:
```
conda install pytorch torchvision torchaudio -c pytorch
```
5. 验证 PyTorch 是否安装成功:
```
python -c "import torch; print(torch.__version__)"
```
如果安装成功,将会输出 PyTorch 的版本号。
阅读全文