要創建一個新的pytorch環境,可以按照以下步驟操作:
打開終端或命令提示符,并運行以下命令來創建一個新的conda環境:
conda create --name <環境名稱> python=<Python版本號>
其中,<環境名稱>
是你想要為新環境指定的名稱,<Python版本號>
是你想要在環境中使用的Python版本號。例如,如果你想要創建一個名為pytorch_env
的環境,并使用Python 3.7版本,可以運行以下命令:
conda create --name pytorch_env python=3.7
激活新創建的環境。運行以下命令來激活剛剛創建的環境:
conda activate <環境名稱>
例如,要激活名為pytorch_env
的環境,可以運行以下命令:
conda activate pytorch_env
安裝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
創建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來使用該環境。記住,在使用該環境時,你需要先激活它。