通过conda安装pytorch
时间: 2024-10-23 14:02:25 浏览: 17
conda安装cuda10.1+cudnn7.6.5,conda安装深度学习gpu环境
通过`conda`安装`PyTorch`是一种常见的包管理方式,它适用于创建独立的环境并管理依赖。以下是使用`conda`安装`PyTorch`的基本步骤:
1. **激活Conda环境**(如果尚未激活,首先需要激活):
```bash
conda activate your_env_name (假设你的环境名为your_env_name)
```
2. **检查当前可用的PyTorch版本**(如果还未安装):
```bash
conda search pytorch
```
3. **安装特定版本的PyTorch**(例如,安装最新稳定版):
```bash
conda install pytorch torchvision torchaudio -c pytorch
```
如果你需要CUDA支持,记得添加` cudatoolkit=version_number`,比如 `cudatoolkit=11.0`。
4. **验证安装**:
```bash
python -m torch
```
这将启动Python交互模式,并显示PyTorch的信息。
阅读全文