溫馨提示×

conda怎么創(chuàng)建新的pytorch環(huán)境

小億
108
2024-01-03 16:27:24
欄目: 編程語言

要創(chuàng)建一個新的pytorch環(huán)境,可以按照以下步驟操作:

  1. 打開終端或命令提示符,并運行以下命令來創(chuàng)建一個新的conda環(huán)境:

    conda create --name <環(huán)境名稱> python=<Python版本號>
    

    其中,<環(huán)境名稱>是你想要為新環(huán)境指定的名稱,<Python版本號>是你想要在環(huán)境中使用的Python版本號。例如,如果你想要創(chuàng)建一個名為pytorch_env的環(huán)境,并使用Python 3.7版本,可以運行以下命令:

    conda create --name pytorch_env python=3.7
    
  2. 激活新創(chuàng)建的環(huán)境。運行以下命令來激活剛剛創(chuàng)建的環(huán)境:

    conda activate <環(huán)境名稱>
    

    例如,要激活名為pytorch_env的環(huán)境,可以運行以下命令:

    conda activate pytorch_env
    
  3. 安裝pytorch和其他必需的依賴庫。在激活的環(huán)境下,運行以下命令來安裝pytorch和其他必需的依賴庫:

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

    其中,<CUDA版本號>是你所使用的CUDA版本號。如果不使用CUDA,可以不提供cudatoolkit參數(shù)。例如,要安裝適用于CUDA 11.1的pytorch,可以運行以下命令:

    conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
    
  4. 創(chuàng)建jupyter內(nèi)核(可選)。如果你希望在新環(huán)境中使用Jupyter Notebook或JupyterLab,可以通過運行以下命令來創(chuàng)建一個新的Jupyter內(nèi)核:

    python -m ipykernel install --user --name=<環(huán)境名稱> --display-name=<顯示名稱>
    

    其中,<環(huán)境名稱>是你的環(huán)境名稱,<顯示名稱>是你在Jupyter中希望顯示的內(nèi)核名稱。例如,要在Jupyter中顯示名為pytorch_env的內(nèi)核,可以運行以下命令:

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

現(xiàn)在,你應該已經(jīng)成功創(chuàng)建了一個新的pytorch環(huán)境。你可以在激活環(huán)境后運行Python腳本或啟動Jupyter Notebook來使用該環(huán)境。記住,在使用該環(huán)境時,你需要先激活它。

0