Ubuntu下PyTorch安裝失敗怎么辦

在Ubuntu下安裝PyTorch時可能會遇到各種問題，以下是一些常見的排查步驟和解決方案：

常見問題及解決方案

1. CUDA未安裝或版本不匹配

   • 檢查CUDA是否安裝：運行 nvcc --version，如果顯示"command not found"，說明CUDA沒有安裝或者環境變量沒有配置好。
   • 安裝合適版本的CUDA：使用以下命令安裝CUDA Toolkit：

     sudo apt update
     sudo apt install nvidia-cuda-toolkit
     

   • 配置環境變量：在~/.bashrc文件末尾添加：

     export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
     export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
     

     然后執行：

     source ~/.bashrc
     

2. PyTorch版本不正確

   • 卸載當前PyTorch并安裝支持CUDA的版本：

     pip uninstall torch torchvision torchaudio
     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
     

3. GPU太舊不被支持

   • 使用舊版PyTorch：安裝支持你GPU的舊版PyTorch。
   • 使用CPU模式：雖然速度慢，但可以運行。
   • 升級硬件。

4. 網絡問題導致安裝失敗

   • 使用清華鏡像源加速安裝：

     pip3 --default-timeout 1000 install --index-url https://mirrors.aliyun.com/pypi/simple torch torchvision torchaudio
     

5. 依賴庫缺失

   • 安裝缺失的依賴庫：例如，如果提示缺少 pandas，可以使用以下命令安裝：

     conda install pandas
     

6. 使用Anaconda安裝

   • 安裝Anaconda或Miniconda：

     wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
     bash Miniconda3-latest-Linux-x86_64.sh
     

   • 創建虛擬環境并激活：

     conda create -n your-virtualenv-name python=3.8
     conda activate your-virtualenv-name
     

   • 使用Conda安裝PyTorch：

     conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
     

7. 驗證安裝

   • 安裝成功后，運行以下命令驗證PyTorch是否成功安裝：

     import torch
     print(torch.__version__)
     print(torch.cuda.is_available())