Ubuntu是一個流行的Linux發行版,廣泛應用于各種計算任務,包括深度學習。PyTorch是一個開源的深度學習框架,由Facebook的AI研究團隊開發,因其動態計算圖和簡潔的API而受到廣泛歡迎。以下是如何在Ubuntu上利用PyTorch社區支持的詳細步驟:
conda create -n pytorch_env python=3.8
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
或者使用pip:
pip install torch torchvision torchaudio
nvidia-smi
命令查看CUDA版本,并參考PyTorch官網的安裝指南選擇合適的版本。通過以上步驟,你可以在Ubuntu上成功安裝和利用PyTorch社區提供的豐富資源,進行深度學習項目的開發和部署。