conda 安装torch包
时间: 2025-01-24 15:10:40 浏览: 76
Conda 是一个流行的包管理器和环境管理器,可以用来安装和管理各种软件包,包括 PyTorch(通常简称为 torch)。以下是使用 conda 安装 torch 包的步骤:
1. **打开终端或命令提示符**:首先,打开你的终端(macOS/Linux)或命令提示符(Windows)。
2. **创建新的 conda 环境(可选)**:为了避免与其他包的依赖关系冲突,建议创建一个新的 conda 环境。你可以使用以下命令创建一个名为 `torch_env` 的新环境,并指定 Python 版本(例如 Python 3.8):
```bash
conda create -n torch_env python=3.8
```
3. **激活新环境**:创建环境后,需要激活它:
```bash
conda activate torch_env
```
4. **安装 torch 包**:使用 conda 安装 torch 包。你可以使用以下命令来安装最新版本的 torch:
```bash
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
如果你有一个支持 CUDA 的 GPU,并且想要安装支持 CUDA 的 torch,可以使用以下命令:
```bash
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
```
请根据你的 CUDA 版本调整 `cudatoolkit` 的版本号。
5. **验证安装**:安装完成后,可以通过进入 Python 解释器并导入 torch 来验证安装是否成功:
```bash
python
```
然后在 Python 提示符下输入:
```python
import torch
print(torch.__version__)
```
如果没有错误,并且输出了 torch 的版本号,说明安装成功。
阅读全文
相关推荐

















