要創(chuàng)建一個新的pytorch環(huán)境,可以按照以下步驟操作:
打開終端或命令提示符,并運行以下命令來創(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
激活新創(chuàng)建的環(huán)境。運行以下命令來激活剛剛創(chuàng)建的環(huán)境:
conda activate <環(huán)境名稱>
例如,要激活名為pytorch_env
的環(huán)境,可以運行以下命令:
conda activate pytorch_env
安裝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
創(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)境時,你需要先激活它。