在Ubuntu上安裝PyTorch時,需要滿足一些基本的依賴項要求。以下是一些常見的依賴項:
python3 --version
如果未安裝Python 3,可以使用以下命令安裝:
sudo apt update
sudo apt install python3 python3-pip
sudo apt install python3-pip
nvidia-smi
如果未安裝CUDA,請訪問NVIDIA官網下載并安裝適合你系統的CUDA版本。
cuDNN(可選):cuDNN是用于深度神經網絡的GPU加速庫。你需要根據安裝的CUDA版本安裝相應的cuDNN版本。例如,如果你安裝了CUDA 11.1,可以從NVIDIA官網下載并安裝cuDNN 8.0.5。
其他依賴項:根據不同的安裝方法和系統配置,可能還需要安裝其他依賴項。例如,在安裝PyTorch之前,可以使用以下命令安裝一些常見的構建工具和庫:
sudo apt-get update
sudo apt-get install -y build-essential curl git-core libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev wget libffi-dev llvm xz-utils libncurses5-dev tk-dev libxmlsec1-dev libxml2-dev
請注意,以上信息可能會隨著PyTorch版本的更新而發生變化。建議在安裝PyTorch之前,訪問PyTorch官網獲取最新的安裝指南。