Linux 中python安装PyTorch
时间: 2023-07-23 10:19:23 浏览: 91
在 Linux 中,可以通过以下步骤安装 PyTorch:
1. 安装 Anaconda:如果还没有安装 Anaconda,可以从官网下载适合自己系统的版本进行安装。
2. 创建新的 Conda 环境:在终端中输入以下命令,创建一个新的 Conda 环境,可以根据自己的需要指定 Python 版本和环境名称:
```
conda create --name pytorch_env python=3.7
```
3. 激活 Conda 环境:输入以下命令来激活创建的 Conda 环境:
```
conda activate pytorch_env
```
4. 安装 PyTorch:输入以下命令来安装 PyTorch,可以根据自己的需要选择合适的版本以及 CPU 或 GPU 版本:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
5. 安装其他依赖项:根据自己的需要安装其他依赖项,例如 NumPy、SciPy、Matplotlib 等。
6. 验证 PyTorch 安装:输入以下命令来验证 PyTorch 是否安装成功:
```
python -c "import torch; print(torch.__version__)"
```
如果输出 PyTorch 的版本号,则说明安装成功。
总之,在 Linux 中安装 PyTorch 需要先安装 Anaconda,然后创建新的 Conda 环境,并在环境中安装 PyTorch 和其他依赖项。
阅读全文