anaconda pytorch环境搭建
时间: 2024-07-19 08:01:15 浏览: 150
如何搭建pytorch环境的方法步骤
5星 · 资源好评率100%
Anaconda是一个流行的Python数据科学平台,它包含了大量的科学计算库,包括PyTorch。以下是创建一个基本的Anaconda PyTorch环境的步骤:
1. **安装 Anaconda**:
如果还没有安装Anaconda,首先访问https://www.anaconda.com/products/distribution 下载适合你的系统版本。
2. **启动 Anaconda Prompt (Windows)** 或者打开终端 (Mac/Linux):
打开命令行界面并运行 `conda create` 命令开始创建新环境。
3. **创建新环境**:
```sh
conda create -n torchenv python=3.9 # 用你想要的Python版本替换3.9
```
这会创建一个名为 "torchenv" 的新环境,并指定Python版本。
4. **激活新环境**:
Windows:
```sh
conda activate torchenv
```
Linux/Mac:
```sh
source activate torchenv
```
5. **安装 PyTorch**:
```sh
conda install pytorch torchvision cudatoolkit=11.0 -c pytorch # 如果你的GPU支持CUDA 11.0,替换为对应版本
```
如果你不需要CUDA加速,可以使用CPU版本 `pip install torch torchvision`.
6. **验证安装**:
完成安装后,你可以通过输入 `python -m torch.__version__` 来检查PyTorch是否已成功安装。
阅读全文