温馨提示×

conda怎么创建新的pytorch环境

小亿
108
2024-01-03 16:27:24
栏目: 编程语言

要创建一个新的pytorch环境,可以按照以下步骤操作:

  1. 打开终端或命令提示符,并运行以下命令来创建一个新的conda环境:

    conda create --name <环境名称> python=<Python版本号>
    

    其中,<环境名称>是你想要为新环境指定的名称,<Python版本号>是你想要在环境中使用的Python版本号。例如,如果你想要创建一个名为pytorch_env的环境,并使用Python 3.7版本,可以运行以下命令:

    conda create --name pytorch_env python=3.7
    
  2. 激活新创建的环境。运行以下命令来激活刚刚创建的环境:

    conda activate <环境名称>
    

    例如,要激活名为pytorch_env的环境,可以运行以下命令:

    conda activate pytorch_env
    
  3. 安装pytorch和其他必需的依赖库。在激活的环境下,运行以下命令来安装pytorch和其他必需的依赖库:

    conda install pytorch torchvision torchaudio cudatoolkit=<CUDA版本号> -c pytorch
    

    其中,<CUDA版本号>是你所使用的CUDA版本号。如果不使用CUDA,可以不提供cudatoolkit参数。例如,要安装适用于CUDA 11.1的pytorch,可以运行以下命令:

    conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
    
  4. 创建jupyter内核(可选)。如果你希望在新环境中使用Jupyter Notebook或JupyterLab,可以通过运行以下命令来创建一个新的Jupyter内核:

    python -m ipykernel install --user --name=<环境名称> --display-name=<显示名称>
    

    其中,<环境名称>是你的环境名称,<显示名称>是你在Jupyter中希望显示的内核名称。例如,要在Jupyter中显示名为pytorch_env的内核,可以运行以下命令:

    python -m ipykernel install --user --name=pytorch_env --display-name="PyTorch Environment"
    

现在,你应该已经成功创建了一个新的pytorch环境。你可以在激活环境后运行Python脚本或启动Jupyter Notebook来使用该环境。记住,在使用该环境时,你需要先激活它。

0