miniconda3安装pytorch
时间: 2025-02-23 07:30:00 浏览: 77
### 安装 PyTorch 使用 Miniconda3
对于希望在 WSL2 上的 Ubuntu 22.04 中通过 Miniconda3 来安装 PyTorch 的用户来说,有几种方法可以实现这一目标。
#### 方法一:仅安装 CPU 版本
如果不需要 GPU 支持,则可以通过如下命令来安装适用于当前设置的 PyTorch 和 torchvision 组件:
```bash
conda install pytorch-cpu torchvision-cpu -c pytorch[^1]
```
此方式适合那些不打算利用图形处理单元加速计算的任务场景。
#### 方法二:带有 CUDA 加速的支持版本
为了获得更好的性能并充分利用 NVIDIA 显卡的能力,在支持 CUDA 的环境中应考虑安装特定版本组合。例如,指定版本号以及来自 `pytorch` 和 `nvidia` 渠道的内容可确保兼容性和稳定性:
```bash
conda install pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 pytorch-cuda=11.8 -c pytorch -c nvidia[^2]
```
这种方法特别推荐给拥有适当硬件配置并且计划执行需要大量运算资源的应用程序开发者们。
#### 配置环境变量生效
完成上述任一步骤之后,请注意按照提示关闭再重新打开终端窗口以便使更改后的路径和其他初始化脚本能够被加载进来。这通常是在首次启动或更新 Conda 后所必需的操作之一:
> For changes to take effect, close and re-open your current shell.[^3]
阅读全文
相关推荐
















