如何用anaconda安装torch
时间: 2024-03-23 22:15:05 浏览: 68
您可以通过以下步骤在 Anaconda 环境中安装 PyTorch:
1. 打开 Anaconda Prompt(Windows)或终端(macOS/Linux)窗口。
2. 输入以下命令创建一个新的虚拟环境并激活它:
```
conda create -n <your_env_name> python=<python_version>
conda activate <your_env_name>
```
其中,`<your_env_name>` 是您想要创建的环境名称,`<python_version>` 是您想要使用的 Python 版本号。例如,我想要创建一个名为 `pytorch-env` 的环境,并使用 Python 3.8,可以输入以下命令:
```
conda create -n pytorch-env python=3.8
conda activate pytorch-env
```
3. 输入以下命令安装 PyTorch:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
如果您想要安装 GPU 版本的 PyTorch,可以输入以下命令:
```
conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch
```
其中,`<your_cuda_version>` 是您所安装的 CUDA 工具包的版本号。您可以通过以下命令来查看当前系统中安装的 CUDA 版本:
```
nvcc --version
```
如果您不确定要安装哪个版本的 PyTorch,请查看 PyTorch 官方网站上的版本说明。
4. 安装完成后,您可以在 Python 解释器中导入 PyTorch 模块并开始使用它:
```
import torch
```
如果没有报错,说明安装成功。
阅读全文