linux下pytorch环境搭建

在Linux下搭建PyTorch环境，可以按照以下步骤进行：

1. 安装Anaconda或Miniconda，可以从官网下载对应版本的安装包进行安装。

2. 创建一个新的conda环境，可以使用以下命令：

   conda create --name pytorch_env python=3.7

其中，pytorch_env是环境名称，python=3.7表示使用Python 3.7版本。

3. 激活新创建的环境，可以使用以下命令：

   conda activate pytorch_env

4. 安装PyTorch和相关依赖，可以使用以下命令：

   conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

其中，cudatoolkit=11.1表示使用CUDA 11.1版本，如果不需要使用GPU加速，可以不加这个参数。

5. 安装其他需要的Python包，可以使用pip命令进行安装，例如：

   pip install matplotlib pandas scikit-learn

这里安装了matplotlib、pandas和scikit-learn三个包。

6. 完成环境搭建后，可以使用以下命令退出当前环境：

   conda deactivate

如果需要删除环境，可以使用以下命令：

   conda remove --name pytorch_env --all

其中，pytorch_env是要删除的环境名称。