conda创建虚拟环境pytorch
时间: 2025-02-18 15:30:01 浏览: 51
### 使用 Conda 创建包含 PyTorch 的虚拟环境
为了创建一个包含 PyTorch 的虚拟环境,可以按照如下方法操作:
#### 安装 Anaconda 或 Miniconda
确保已经安装了 Anaconda 或者 Miniconda。这一步骤提供了 `conda` 工具来管理包和环境。
#### 创建新的 Python 虚拟环境并指定版本
通过下面命令创建一个新的 Python 3.6 版本的虚拟环境名为 `pytorch_env`:
```bash
conda create -n pytorch_env python=3.6
```
激活新创建的环境以便后续在其内部安装其他依赖项[^1]:
```bash
conda activate pytorch_env
```
#### 添加 PyTorch 及其相关库到环境中
根据所需平台(Windows, macOS, Linux),以及是否需要 GPU 支持,选择合适的 PyTorch 安装方式。对于 CPU-only 版本,在已激活的目标环境下执行以下指令即可完成 PyTorch 和 torchvision 库的安装:
```bash
conda install pytorch torchvision cpuonly -c pytorch
```
如果目标机器有 NVIDIA 显卡并且希望利用 CUDA 加速,则应替换上述命令中的 `-cpuonly` 参数为具体的 CUDA 版本号,例如针对 CUDA 10.2 的情况可使用:
```bash
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
```
这样就可以成功设置好带有特定配置选项的 PyTorch 开发环境了[^2]。
当不再需要当前工作区时,可以通过运行 `conda deactivate` 来退出该虚拟环境。
阅读全文
相关推荐


















