在ubuntu上安装pytorch
时间: 2023-10-18 07:22:43 浏览: 75
可以通过以下步骤在 Ubuntu 上安装 PyTorch:
1. 确认已经安装了 Python。可以在终端输入以下命令进行检查:
```
python3 --version
```
2. 安装 PyTorch 的依赖项。在终端中输入以下命令:
```
sudo apt-get update
sudo apt-get install python3-pip python3-dev python3-venv
```
3. 创建一个虚拟环境。在终端输入以下命令:
```
python3 -m venv myenv
```
该命令将在当前目录下创建一个名为“myenv”的虚拟环境。
4. 激活虚拟环境。在终端中输入以下命令:
```
source myenv/bin/activate
```
5. 安装 PyTorch。在终端中输入以下命令:
```
pip3 install torch torchvision torchaudio
```
如果需要安装特定版本的 PyTorch,可以使用以下命令:
```
pip3 install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio==0.7.2 -f https://download.pytorch.org/whl/cpu/torch_stable.html
```
6. 验证 PyTorch 是否安装成功。在终端中输入以下命令:
```
python3 -c "import torch; print(torch.__version__)"
```
如果输出 PyTorch 的版本号,则表示安装成功。
7. 退出虚拟环境。在终端中输入以下命令:
```
deactivate
```
这样就可以退出虚拟环境了。
希望这些步骤对你有所帮助!
阅读全文