在linux的conda配置pytorch库
时间: 2024-11-09 18:20:38 浏览: 28
在Linux远程服务器上设置PyTorch的GPU版本
在Linux环境下使用Conda管理PyTorch库,你需要按照以下步骤操作:
1. **安装Miniconda**:首先,从Anaconda官网下载Miniconda的适合Linux的版本,然后按照提示安装。
```bash
# 下载Miniconda
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
# 创建一个新的环境并安装
bash Miniconda3-latest-Linux-x86_64.sh -b
```
2. **创建新环境并激活**:创建一个名为`pytorch_env`的新环境,并安装PyTorch。
```bash
conda create --name pytorch_env python=3.7
source activate pytorch_env
```
3. **安装PyTorch**:使用`conda install`命令安装PyTorch,可以选择CPU版本或CUDA版本,取决于你的系统是否支持GPU。
```bash
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```
如果你的系统没有显卡或者不需要CUDA,可以替换为`pip install torch torchvision torchaudio`。
4. **检查安装**:确认PyTorch已经安装成功。
```bash
python -c "import torch; print(torch.__version__)"
```
5. **如果需要其他依赖**:如果有其他深度学习相关的包需求,可以直接在环境中安装。
```bash
conda install numpy scipy pandas
```
阅读全文