anaconda配置pytorch环境ubantu
时间: 2024-10-08 21:07:25 浏览: 163
Anaconda是一个流行的Python数据科学平台,它包含了包管理器`conda`以及大量的科学计算库,包括PyTorch。在Ubuntu系统上配置PyTorch环境通常分为几个步骤:
1. **安装Miniconda** (轻量级版本) 或者 **Anaconda**:
- 打开终端并访问Miniconda官网下载适用于Ubuntu的安装脚本:https://docs.conda.io/en/latest/miniconda.html#linux
- 运行脚本并按照提示操作,选择安装路径,然后一路回车。
2. **创建新环境**:
```
conda create --name my_pytorch_env python=3.8
```
这里`my_pytorch_env`是新环境的名称,可以根据需要自定义,`python=3.8`指定了Python版本。
3. **激活环境**:
```
conda activate my_pytorch_env
```
4. **安装PyTorch**:
```
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
```
如果你的电脑支持CUDA GPU,你需要指定对应的CUDA版本。如果没有GPU,可以换成`cpu`版本。
5. **检查安装**:
运行`python -c "import torch; print(torch.__version__)"`,确认PyTorch已经安装成功并且能正常导入。
6. **其他依赖**:
根据你的项目需求,可能还需要安装其他如TensorFlow、NumPy等库,用类似的方式添加到环境中。
阅读全文
相关推荐
















