Skip to main content

M40 安装cuda和torch

  1. 驱动
    1. sudo add-apt-repository ppa:graphics-drivers/ppa
    2. sudo apt update
    3. sudo apt install nvidia-driver-550

  2. CUDA 11.8
    1. https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local
    2. sudo sh cuda*
  3. Pytorch
    1. https://pytorch.org/get-started/previous-versions/
    2. pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu118