ubuntu安装PyTorch 1.8和其他依赖项
时间: 2024-03-25 11:40:40 浏览: 158
安装 PyTorch 1.8 和其他依赖项的步骤如下:
1. 安装 Anaconda:在 Anaconda 官网下载对应的安装包,按照提示进行安装。
2. 创建新的虚拟环境:在终端中输入以下命令创建一个名为 "pytorch" 的新虚拟环境。
```
conda create -n pytorch python=3.8
```
3. 激活虚拟环境:在终端中输入以下命令激活虚拟环境。
```
conda activate pytorch
```
4. 安装 PyTorch:在终端中输入以下命令安装 PyTorch。
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
5. 安装其他依赖项:在终端中输入以下命令安装其他依赖项。
```
conda install matplotlib numpy pandas scipy scikit-learn jupyter notebook
```
6. 验证 PyTorch 安装:在 Python 终端中输入以下命令验证 PyTorch 是否成功安装。
```
import torch
print(torch.__version__)
```
如果输出版本号为 1.8,则说明 PyTorch 安装成功。
注意事项:
- 如果你的 GPU 是 NVIDIA 的,可以使用 cudatoolkit 参数安装 GPU 版本的 PyTorch。
- 如果你没有 GPU,可以省略 cudatoolkit 参数安装 CPU 版本的 PyTorch。
- 如果你的系统不是 Ubuntu,可以根据需要修改安装命令。
阅读全文