linux conda 安装pytorch
时间: 2023-10-29 14:53:21 浏览: 119
使用anaconda安装pytorch的实现步骤
您可以使用 conda 命令来安装 PyTorch。首先,打开终端或命令行窗口,并确保已经安装了 conda。然后,执行以下命令来创建一个新的 conda 环境并安装 PyTorch:
```shell
conda create -n myenv
conda activate myenv
conda install pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch
```
请将 `myenv` 替换为您想要使用的环境名称,`xx.x` 替换为您系统上可用的 CUDA 版本。如果您不使用 GPU,可以省略 `cudatoolkit` 部分。
执行完毕后,PyTorch 将会安装在您的 conda 环境中。您可以通过执行以下命令来验证安装是否成功:
```shell
python
import torch
print(torch.__version__)
```
如果没有报错并打印出版本号,则表示安装成功。
阅读全文